py 数组按特定行列拆分

可以使用NumPy的split函数来按特定行列拆分数组。下面是一个示例代码：

import numpy as np

# 创建一个 4x6 的数组
arr = np.arange(24).reshape((4, 6))
print(arr)

# 按行拆分为两个数组
arr_split = np.split(arr, 2, axis=0)
print(arr_split)

# 按列拆分为三个数组
arr_split = np.split(arr, [2, 4], axis=1)
print(arr_split)

输出：

[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]
 [12 13 14 15 16 17]
 [18 19 20 21 22 23]]
[array([[0, 1, 2, 3, 4, 5],
       [6, 7, 8, 9, 10, 11]]), array([[12, 13, 14, 15, 16, 17],
       [18, 19, 20, 21, 22, 23]])]
[array([[ 0,  1],
       [ 6,  7],
       [12, 13],
       [18, 19]]), array([[ 2,  3],
       [ 8,  9],
       [14, 15],
       [20, 21]]), array([[ 4,  5],
       [10, 11],
       [16, 17],
       [22, 23]])]

这里使用`np.split(arr, 2, axis=0)`按行拆分为两个数组，使用`np.split(arr, [2, 4], axis=1)`按列拆分为三个数组。`np.split()`函数的第二个参数为拆分点，可以是一个整数、一个列表或一个数组，表示从哪些位置拆分数组。在第二个例子中，`[2, 4]`表示在第2列和第4列处拆分数组。