numpy.array_split¶

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

Split an array into multiple sub-arrays.

Please refer to thesplit documentation. The only differencebetween these functions is thatarray_split allowsindices_or_sections to be an integer that doesnot equallydivide 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.])]