沿着一定的维度将一个numpy数组附加到另一个数组上

Question

存在两个numpy.ndarray，如A的形状为(5,3,2),B的形状为(5,3)。

A = np.random.rand(5,3,2)
B=np.random.rand(5,3)  #----- there was typo here

我想将B追加到A中，并使结果数组C与形状(5,3,3)我试图使用np.contaenate，但它不起作用。

c=np.concatenate((a,b),axis=2) 

Answer 1

import numpy as np
a = np.random.rand(5,3,2)
b = np.random.rand(5,3)
b = b[..., np.newaxis] # or np.expand_dims(b, axis=2) or b.reshape(b.shape+(1,))
print(b) # (5, 3, 1)
c = np.append(a, b, axis=2)
print(c.shape) # (5, 3, 2)