Tuesday, January 27, 2009

Snippets: Creating a matrix in python

There's two easy ways to do this:

Looping:
  1. def matrix(rownumber,columnumber):  
  2.    matrix = []  
  3.    for i in range(rownumber):  
  4.        a = [0]*columnumber  
  5.        matrix.append(a)  
  6.    return matrix  


List comprehensions:
  1. def matrix(rownumber, columnumber):  
  2.     return [[0]*columnumber for x in range(rownumber)]  

No comments:

Post a Comment

Please, be polite and constructive.