numpy中array的分割

import numpy as np

a = np.arange(12).reshape((3, 4))
print(a)

print(np.split(a, 2, axis=1))  # 纵向分割

print(np.array_split(a, 2, axis=0))  # 并不等量的强行分割

# 纵横向分割
print(np.vsplit(a, 3))
print(np.hsplit(a, 2))

学习自莫凡Python