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
?
Advertisement
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]]