纵向分割:
np.split(A, 2, axis=1)
横向分割:
np.split(A, 3, axis=0)
以上只能等量分割,如果无法整除,将出错。
如果需要不等量分割:
np.array_split(A, 3, axis=1)
array_split(A,n,axis=1)解释为:纵向分割为n份,其中每份的大小以如下的计算方式计算:
纵向长度为L,n对L求余为a,n整除L为b,那么分割构成为:
a份b+1大小的块,其余全部为b大小的块。
print(np.hsplit(A, 2)) #等于 print(np.split(A, 2, axis=1))
print(np.vsplit(A, 3)) #等于 print(np.split(A, 3, axis=0))