python - 将数组添加到数组的维度

Question

我有 2 个 numpy 数组img，mask我想组合成一个数组。数组的形状如下： image.shape = (512, 366, 3)和mask.shape = (512, 366)。我希望最终数组的形状为(512, 366, 4)，使得mask数组占据第 4 维。

请问，实现这一目标的最佳方法是什么？

标签： pythonarraysnumpyappend

Answer 1

朱利安的建议有效：

from numpy import stack
new_image = dstack(img, mask)