Numpy Split на куски одинакового размера
def build_chunks(arr, chunk_size, axis=1):
return np.split(arr,
range(chunk_size, arr.shape[axis], chunk_size), axis=axis)
Adorable Anaconda