使用2d numpy mask np.where寻址3d numpy数组（pythonic ??）

Question

说我有一个图像数组：

raster.shape => (3,100,100)

我为红色饱和的所有位置生成一个蒙版：

mask = np.where(raster[0,:,:] == 255)

我想修改这些位置的一部分...我可以这样：

raster[:,mask[0],mask[1]] = 0

有没有更Python的方式...还是最好的选择？

Answer 1

我会直接使用口罩，例如

raster[:, raster[0] == 255] = 0

这应该相当于您写的内容。