numpy的array分割

import numpy as np 

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

print(a)

print("纵向分割（按照列进行分割）：\n", np.split(a, 2, axis=1))
print("横向分割（按照行进行分割）：\n", np.split(a, 3, axis=0))

print("不等量的分割：\n", np.array_split(a, 3, axis=1))
print("简化的纵向分割：\n", np.vsplit(a, 3))
print("简化的横向分割：\n", np.hsplit(a, 2))