numpy.split()

numpy.split

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

Split an array into multiple sub-arrays.

Parameters:

ary : ndarray

Array to be divided into sub-arrays.

indices_or_sections : int or 1-D array

If indices_or_sections is an integer, N, the array will be divided into N equal arrays along axis. If such a split is not possible, an error is raised.

If indices_or_sections is a