Python: Numpy: 分割

纵向分割:

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))

猜你喜欢

转载自blog.csdn.net/lcqin111/article/details/82662803