将数组添加到数组的维度

时间:2019-01-24 23:26:56

标签: python arrays numpy append

我有2个numpy数组imgmask,我想合并为一个数组。数组的形状如下: image.shape = (512, 366, 3)mask.shape = (512, 366)。我希望最终数组的形状为(512, 366, 4),这样mask数组占据第4维。

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

1 个答案:

答案 0 :(得分:0)

朱利安作品的建议:

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