x = np.arange(8.0) np.array_split(x, 3) # returns [array([0., 1., 2.]), array([3., 4., 5.]), array([6., 7.])]