numpy convert 1d ke 2d

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