numpy

numpy.split

numpy.split(ary, indices_or_sections, axis=0)

>>> x = np.arange(9.0)>>> np.split(x, 3)[array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.,  8.])]>>> x = np.arange(8.0)>>> np.split(x, [3, 5, 6, 10])[array([ 0.,  1.,  2.]), array([ 3.,  4.]), array([ 5.]), array([ 6.,  7.]), array([], dtype=float64)]

numpy.array_split

numpy.array_split(ary, indices_or_sections, axis=0)

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