numpy 分割

import numpy as np

# 分割
A = np.arange(12).reshape((3,4))
print(A)

# 纵向分割
print(np.split(A, 2, axis=1))
print('***************************')
# 不等分割
print(np.array_split(A, 3, axis=1))
print('***************************')
# 纵向分割
print(np.vsplit(A,3))
print('***************************')
# 横向分割
print(np.hsplit(A, 2))

结果：
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
[array([[0, 1],
[4, 5],
[8, 9]]), array([[ 2, 3],
[ 6, 7],
[10, 11]])]

---

[array([[0, 1],
[4, 5],
[8, 9]]), array([[ 2],
[ 6],
[10]]), array([[ 3],
[ 7],
[11]])]

---

[array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]]), array([[ 8, 9, 10, 11]])]

---

[array([[0, 1],
[4, 5],
[8, 9]]), array([[ 2, 3],
[ 6, 7],
[10, 11]])]