matriz separada a lo largo del eje

np.split(arr, indices_or_sections, axis=0)
Real Raccoon