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