Numpy.split

NumPy中的split函数用于将一个数组沿指定轴分割成多个子数组。它的语法如下：

numpy.split(ary, indices_or_sections, axis=0)

其中，参数ary是要分割的数组，indices_or_sections可以是一个整数或者一维数组，用于指定分割点的位置，axis表示要沿着哪个轴进行分割。

如果indices_or_sections是一个整数n，则将数组ary分割成n个相等大小的子数组。如果indices_or_sections是一个一维数组，则根据数组中的元素值来分割ary。

下面是一些示例：

1. 将数组ary沿着水平方向（轴0）分割成3个相等大小的子数组：

import numpy as np

ary = np.array([1, 2, 3, 4, 5, 6])
result = np.split(ary, 3, axis=0)
print(result)

输出结果为：

[array([1, 2]), array([3, 4]), array([5, 6])]

2. 将数组ary沿着垂直方向（轴1）分割成两个子数组，分割点为索引1：

import numpy as np

ary = np.array([[1, 2, 3], [4, 5, 6]])
result = np.split(ary, [1], axis=1)
print(result)

输出结果为：

[array([,
]), array([[2, 3],
[5, 6]])]

3. 将数组ary沿着垂直方向（轴1）分割成三个子数组，分割点为索引2和索引4：

import numpy as np

ary = np.array([[1, 2, 3, 4, 5, 6]])
result = np.split(ary, [2, 4], axis=1)
print(result)

输出结果为：

[array([[1, 2],
[4, 5]]), array([[3, 4],
[6, 7]]), array([[5, 6]])]