Python: Matrix w/ Sequential Values

Question

I need to do the following:

Create a variable result as an array of length m.
For each row i from 0 to m-1:

a. Set result[i] to a new array of length n.

b. For each column j in row i, set that element to (i*n)+j.
When you’ve computed all the elements, return result.

So far, I have the following:

     def f2(m, n):
         result = [""] * m
         for i in range(0, m):
             result[i] = [""] * n   
         for i in result:
             for j in i:
                 result.append(i*n+j)
         return result

This, however, just results in an error.

Any help/suggestions? Thank you

Do you actually need to do the sequence of things you've listed, or do you simply need to produce an m-by-n nested list whose elements are the integers from 0 to m*n, in sequence? — user2357112
– user2357112, Commented Dec 1, 2013 at 7:15

bcorso · Accepted Answer · 2013-12-01 07:34:45Z

1

Use list comprehension:

def f2(m, n):
    return [[i*n+j for j in range(n)] for i in range(m)]

edited Dec 1, 2013 at 7:34

answered Dec 1, 2013 at 7:29

bcorso

47.5k10 gold badges66 silver badges76 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

aIKid · Accepted Answer · 2013-12-01 07:23:13Z

0

Here you go:

def f2(m, n):
    #First, let's create a list with the size of m, and
    #each row with the size of n.
    result = [[0 for _ in range(n)] for _ in range(m)]
    #now, you can loop through it easily.
    for i in range(m): #outer list
         for j in range(n): #inner list
             result[i][j] = i*n+j #put it there :)
    return result

Hope this helsp!

edited Dec 1, 2013 at 7:23

answered Dec 1, 2013 at 7:18

aIKid

28.5k5 gold badges41 silver badges65 bronze badges

4 Comments

Deelaka Over a year ago

TypeError: object of type 'int' has no len()

aIKid Over a year ago

@KDawG Whoops, fixed.

user2357112 Over a year ago

You have the variables m and n. Why the awkward range-len?

aIKid Over a year ago

@user2357112 You've got a point.. Hahaha sorry, i'll change that.

user2357112 · Accepted Answer · 2013-12-01 07:24:16Z

for i in result:
    for j in i:

In this nested loop, i iterates over the rows of result and j iterates over the cell contents. Thus, in the loop body:

        result.append(i*n+j)

this line attempts to multiply a row by n, add j (a string) to that, and append the output of that to result. No part of that is what you want to do. You could fix this by iterating over the indices and assigning to elements instead of appending:

for i in range(m):
    for j in range(n):
        result[i][j] = i*n+j

but it's easier to build the whole array the way you want it with range instead of initializing it with garbage and assigning the elements:

def f2(m, n):
    return [range(n*i, n*i+n) for i in xrange(m)]

or if you're on Python 3:

def f2(m, n):
    return [list(range(n*i, n*i+n)) for i in range(m)]

or if list comprehensions are strange and unreadable to you:

def f2(m, n):
    result = []
    for i in range(m):
        result.append(list(range(n*i, n*i+n)))
    return result

Collectives™ on Stack Overflow

Python: Matrix w/ Sequential Values

3 Answers 3

Comments

4 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

4 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related