Python: Numpy: 合并

vstack(A,B)对AB进行列方向合并

hstack(A, B)对AB进行行方向的合并

A = np.array([1,1,1])
B = np.array([2,2,2])
         
print(np.vstack((A,B)))    # vertical stack
"""
[[1,1,1]
 [2,2,2]]
"""
D = np.hstack((A,B))       # horizontal stack

print(D)
# [1,1,1,2,2,2]

print(A.shape,D.shape)
# (3,) (6,)

 np.newaxis():

print(A[np.newaxis,:])
# [[1 1 1]]
print(A[np.newaxis,:].shape)
# (1,3)
print(A[:,np.newaxis])
"""
[[1]
[1]
[1]]
"""
print(A[:,np.newaxis].shape)
# (3,1)

第一个参数位,增加行的维度,第二个参数位增加列的维度

np.concatenate():

C = np.concatenate((A,B,B,A),axis=0)

axis=0时,为列方向合并,一行一行的堆加,与vstack()一样,axis=1时,为行方向的合并,与hstack()一样

猜你喜欢

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