NumPy陣列基礎與計算(四):Numpy陣列的拼接和分裂
阿新 • • 發佈:2018-12-20
前文的操作畢竟只是針對於單一陣列,但是有時候我們需要對多個數組進行操作,比如說拼接和分裂
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]]