python - 如何附加 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))