numpy的array分割

import numpy as np

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

print(np.split(A,2,axis=1))
print(np.split(A,3,axis=0))
#split不能进行不等分割
#print(np.split(A,3,axis=1))
print(np.array_split(A,3,axis=1))
#纵向分割
print(np.vsplit(A,3))
#横向分割

print(np.hsplit(A,2))
print(np.array_split(A,1,axis=1))