如何附加numpy数组？

Question

我想追加如下所示的 numpy 数组

A: [[1,2,3],[2,3,1],[1,4,2]] 
B: [[1,3,3],[3,3,1],[1,4,5]]
A+B = [[[1,2,3],[2,3,1],[1,4,2]],
       [[1,3,3],[3,3,1],[1,4,5]]]

我怎样才能做到这一点？

====================

从注释中复制的代码，并为清晰起见进行了格式化：

X = np.empty([54, 7]) 
for seq in train_set: 
    print(seq) 
    temp = dp.set_xdata(seq) #make 2d numpy array 
    print(temp.shape) 
    X = np.row_stack((X[None], temp[None])) 
X = np.delete(X, 0, 0) 
print("X: ",X) 

ValueError: all the input arrays must have same number of dimensions. 

Answer 1

一种方法是np.vstack在这些数组的 3D 扩展版本上使用-

np.vstack((A[None],B[None]))

另一种方式np.row_stack（功能相同np.vstack） -

np.row_stack((A[None],B[None]))

同样与np.concatenate-

np.concatenate((A[None],B[None])) # By default stacks along axis=0

另一种方法是使用np.stack并指定堆叠轴，即axis=0或跳过它，因为这是默认的堆叠轴 -

np.stack((A,B))