python - 将 2D 数组 append 到 3D 数组，扩展三维

Question

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

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

我试过 np.append(A,B) 但它不保持维度，而 axis 选项导致 ValueError: 所有输入数组必须具有相同的维数。

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)