python - 在第 4 维连接两个 numpy 数组

Question

我有两个具有三个维度 (3 x 4 x 5) 的 numpy 数组，我想将它们连接起来，因此结果有四个维度 (3 x 4 x 5 x 2)。在 Matlab 中，这可以使用 cat(4, a, b) 来完成，但在 Numpy 中则不行。

例如:

a = ones((3,4,5))
b = ones((3,4,5))
c = concatenate((a,b), axis=3) # error!

澄清一下，我希望 c[:,:,:,0] 和 c[:,:,:,1] 对应原来的两个数组.

Answer 1

怎么样

c = np.stack((a,b), axis=3)