Numpy Split a trozos de igual tamaño
def build_chunks(arr, chunk_size, axis=1):
return np.split(arr,
range(chunk_size, arr.shape[axis], chunk_size), axis=axis)
Adorable Anaconda