Numpy converte 1d a 2d

a = np.arange(6).reshape((3, 2))
>>> a
array([[0, 1],
       [2, 3],
       [4, 5]])
Careful Caterpillar