我有一个 256×256 像素的 2D 蒙版，像素值为 0 或 255，以及一个匹配的灰度图像。两者都有形状 (256, 256)。
我想在灰度图像上应用蒙版，因此生成的图像仅包含原始灰度图像的像素，其中相应的蒙版值为 255(应保留原始灰度像素值)。
我无法使用
img_clean = img_grayscale[mask]
因为它向数组添加了一个新维度(如 numpy 的文档中所述)。
我该如何解决？
请您参考如下方法:
只需添加第三个选项并修改您的灰度图像就地:
img_grayscale[mask == 255] = 0
PS:如果您不想就地修改，您可以只创建一个副本。
