1. 程式人生 > >numpy的array分割

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