numpy.array_split — NumPy v1.13 Manual (original) (raw)

numpy. array_split(ary, indices_or_sections, axis=0)[source]¶

Split an array into multiple sub-arrays.

Please refer to the split documentation. The only difference between these functions is that array_split allows_indices_or_sections_ to be an integer that does not equally divide the axis.

See also

split

Split array into multiple sub-arrays of equal size.

Examples

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