Python Nested List Comprehensions to create a matrix

Viewed 20394

I'm new to list comprehensions and would like to replicate the following code below in for loops into a nested list comprehension.

 master = []
 inner = []
 for x in range(1,8):
     inner = []
     for y in range(1,x+1):
         inner.append(str(x*y).rjust(2))

     master.append(inner)

for m in master:
    print ' '.join(m)

The output looks like this:

  1
  2  4
  3  6  9
  4  8 12 16
  5 10 15 20 25
  6 12 18 24 30 36
  7 14 21 28 35 42 49

I was thinking something like:

 test =  [  
    [str(x*y).rjust(2) for x in range(1,8)] 
    for y in range(1,x+1) 
 ] 
 for t in test:
     print ' '.join(t)

But I get a traceback error:

 Traceback (most recent call last):
 File "python", line 3, in <module>
 NameError: name 'x' is not defined

Any python experts care to point me in the right direction?

Thank you in advance!

3 Answers

Python 3

print("\n".join([" ".join([str(x*y) for y in range(1,x+1)]) for x in range(1,8) ]))

Python 2

print "\n".join([" ".join([str(x*y) for y in range(1,x+1)]) for x in range(1,8) ])

Two Step Process.

Join the list by space viz. output '1', '2 4', '3 6 9' etc

Join the resultant list by '\n' and print the result

Related