array的分割

import numpy as np
arr1 = np.arange(12).reshape(3, 4)
print(arr1)
arr2, arr3 = np.split(arr1, 2, axis=1)  # 切分成两个部分，按水平方向
print()
print(arr2)
print()
print(arr3)
arr4, arr5, arr6 = np.split(arr1, 3, axis=0)  # 切分成3个部分，按垂直方向
print(arr4)
print(arr5)
print(arr6)
print()
# arr2, arr3, arr4 = np.split(arr1, 3, axis=1)  # 切分成3个部分，按水平方向(出错)
arr7, arr8, arr9 = np.array_split(arr1, 3, axis=1)  # 切分成3个不等部分，按水平方向(出错)
print(arr7)
print(arr8)
print(arr9)
print()
arrv1, arrv2, arrv3 = np.vsplit(arr1, 3)  # 垂直方向进行分割
print(arrv1)
print(arrv2)
print(arrv3)
print()
arrh4, arrh5 = np.hsplit(arr1, 2)  # 水平方向进行分割
print(arrh4)
print(arrh5)

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

[[0 1]
 [4 5]
 [8 9]]

[[ 2  3]
 [ 6  7]
 [10 11]]
[[0 1 2 3]]
[[4 5 6 7]]
[[ 8  9 10 11]]

[[0 1]
 [4 5]
 [8 9]]
[[ 2]
 [ 6]
 [10]]
[[ 3]
 [ 7]
 [11]]

[[0 1 2 3]]
[[4 5 6 7]]
[[ 8  9 10 11]]

[[0 1]
 [4 5]
 [8 9]]
[[ 2  3]
 [ 6  7]
 [10 11]