1. 程式人生 > >NumPy陣列基礎與計算(四):Numpy陣列的拼接和分裂

NumPy陣列基礎與計算(四):Numpy陣列的拼接和分裂

前文的操作畢竟只是針對於單一陣列,但是有時候我們需要對多個數組進行操作,比如說拼接和分裂

1、陣列的拼接

連線兩個陣列主要由np.concatenate,np.vstack和np.hstack實現

x=np.array([1,2,3])
y=np.array([3,2,1])
np.concatenate([x,y]) #利用np.concatenate對一維陣列進行拼接

array([1, 2, 3, 3, 2, 1])
z=[99,99,99]
np.concatenate([x,y,z]) #利用np.concatenate對三個以上的一維陣列進行拼接
array([ 1,  2,  3,  3,  2,  1, 99, 99, 99])
grid=np.array([[1,2,3],
               [4,5,6]])
np.concatenate([grid,grid]) # 利用np.concatenate進行二維陣列拼接         

 array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 3],
       [4, 5, 6]])    
       
np.concatenate([grid,grid],axis=0) # 當axis=0時沿著第二個軸進行拼接
array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 3],
       [4, 5, 6]])    
 
 np.concatenate([grid,grid],axis=1) # 當axis=1時沿著第二個軸進行拼接
 array([[1, 2, 3, 1, 2, 3],
       [4, 5, 6, 4, 5, 6]])

沿著固定維度去處理陣列時,使用np.vstack(垂直棧)和np.hstack(水平棧)實現會更加的簡潔

x=np.array([1,2,3])
grid=np.array([[4,5,6],[7,8,9]]) #np.vstack(垂直棧)
np.vstack([x,grid])

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
x=np.array([[1,2,3],[3,2,1]])
grid=np.array([[4,5,6],[7,8,9]])
np.hstack([x,grid]) # np.hstack(水平棧)

array([[1, 2, 3, 4, 5, 6],
       [3, 2, 1, 7, 8, 9]])

2、陣列的分裂

與拼接相反的就是分裂,分裂可以通過np.split(),np.hsplit()和np.vsplit()來實現,均是向函式中傳遞一個維度引數:

x=np.array([1,2,3,99,99,3,2,1])
x1,x2,x3=np.split(x,[3,5])# 傳遞一個索引列表進行傳遞
x1,x2,x3

(array([1, 2, 3]), array([99, 99]), array([3, 2, 1]))
grid=np.arange(16).reshape(4,4)
grid

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

upper,lower=np.vsplit(grid,[2])
print(upper)
print(lower)

[[0 1 2 3]
 [4 5 6 7]]
[[ 8  9 10 11]
 [12 13 14 15]]

left,right=np.hsplit(grid,[2])
print(left)
print(right)

[[ 0  1]
 [ 4  5]
 [ 8  9]
 [12 13]]
[[ 2  3]
 [ 6  7]
 [10 11]
 [14 15]]