python np.split怎么用

`np.split`函数可以将一个数组沿着指定的轴分割成多个子数组。

下面是一个示例代码：

import numpy as np

# 创建一个形状为 (6, 3) 的二维数组
arr = np.arange(18).reshape((6, 3))
print("原始数组：")
print(arr)

# 沿着行轴分割成三个子数组
sub_arrs = np.split(arr, 3, axis=0)
print("分割后的子数组：")
for sub_arr in sub_arrs:
    print(sub_arr)

输出结果如下：

原始数组：
[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]
 [12 13 14]
 [15 16 17]]
分割后的子数组：
[[0 1 2]
 [3 4 5]]
[[6 7 8]
 [9 10 11]]
[[12 13 14]
 [15 16 17]]

`np.split`的参数说明：

- `ary`：需要分割的数组
- `indices_or_sections`：可以是一个整数，表示要分成几个子数组；也可以是一个一维数组，指定分割点的位置；还可以是一个元组，指定在哪些位置进行分割。例如，若 `indices_or_sections=3`，则将数组沿着指定轴分成3个子数组。
- `axis`：指定在哪个轴上进行分割。

注意事项：

- 分割后的子数组的形状应该是一致的。如果不能等分，会抛出异常。
- 分割后的子数组是原始数组的视图，即共享内存。如果需要避免修改原始数组的同时修改分割后的子数组，可以使用 `np.array` 函数创建一个新的数组。