Create a 2D list out of 1D list

I am a bit new to Python and I want to convert a 1D list to a 2D list, given the width and length of this matrix.

Say I have a list=[0,1,2,3] and I want to make a 2 by 2 matrix of this list.

How can I get matrix [[0,1],[2,3]] width=2, length=2 out of the list?

Answer

Try something like that:

In [53]: l = [0,1,2,3]

In [54]: def to_matrix(l, n):
    ...:     return [l[i:i+n] for i in xrange(0, len(l), n)]

In [55]: to_matrix(l,2)
Out[55]: [[0, 1], [2, 3]]

JavaScript
​x
 
In [53]: l = [0,1,2,3]
​
In [54]: def to_matrix(l, n):
    ...:     return [l[i:i+n] for i in xrange(0, len(l), n)]
​
In [55]: to_matrix(l,2)
Out[55]: [[0, 1], [2, 3]]
​

Advertisement

Answer