Пример функции разделения Python Numpy
# welcome to softhunt.net
# import numpy
import numpy as np
array = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
# using numpy.split() method
softhunt = np.split(array, [3, 5, 6, 10])
print(softhunt)
Outrageous Ostrich