data = [[1,2,6],[4,5,6],[7,8,9]]
split_array = np.array_split(data, int(1 + 1))
print(split_array)
#
[array([[1, 2, 6],
[4, 5, 6]]), array([[7, 8, 9]])]
data = [[1,2,6],[4,5,6],[7,8,9]]
split_array = np.array_split(data, int(1 + 1))
print(split_array)
#
[array([[1, 2, 6],
[4, 5, 6]]), array([[7, 8, 9]])]