将 2D 数组附加到 3D 数组，扩展第三维

Question

我有一个数组A具有形状(480, 640, 3)和阵列B具有形状(480, 640)

如何将这两个附加为一个形状为(480, 640, 4)数组？

我试过np.append(A,B)但它没有保留维度，而axis选项导致ValueError: all the input arrays must have same number of dimensions 。

Answer 1

使用dstack ：

>>> np.dstack((A, B)).shape
(480, 640, 4)

这将处理数组具有不同维数的情况，并沿第三轴堆叠数组。

否则，要使用append或concatenate ，您必须自己制作三维B并指定要加入它们的轴：

>>> np.append(A, np.atleast_3d(B), axis=2).shape
(480, 640, 4)

Answer 2

使用np.stack应该可以工作，但问题是两个数组都应该是 2D 形式。

np.stack([A,B])