“Отдельный массив вдоль оси” Ответ

Отдельный массив вдоль оси

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