将二维数组附加到三维数组，扩展三维

Question

我有一个形状为(480, 640, 3)的数组A，还有一个形状为(480, 640)的数组B。

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

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

Answer 1

使用^{}：

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

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

否则，要使用append或concatenate，您必须使B成为三维，并指定要连接它们的轴：

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