Python 附加维numpy数组

我有两个numpy数组矩阵A和B，长度为32，形状为（32,32,3）。我想把它们组合成一个新数组，这样我的新数组的维数是（2,32,32,3）

---

使用np。连接正在抛出错误。

使用

np.stack

def dim0_stack(*arrays):
    return np.stack(arrays, axis = 0)

另一种方法是：

a = np.random.randn(32, 32, 3)
b = np.random.randn(32, 32, 3)
c = np.concatenate([np.expand_dims(a,0), np.expand_dims(b, 0)], axis=0)
print(c.shape)

---

因为您提到了使用

串联

，所以我想向您展示一下如何使用它。

另一种更为文字化的方法

result = np.zeros((2, A.shape[0], A.shape[1], A.shape[2]))
result[0, :, :, :] = A
result[1, :, :, :] = B

---

串联

会抛出什么错误？我们需要细节