如何将掩码应用于保持形状和检索值的 numpy 数组？

Question

我正在创建一个蒙版（布尔数组）的图像分割。我现在想在保持图像阵列形状的同时提取遮罩区域的成像数据。

由于图像数据太大，这里是我的问题的简化形式：

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([False, False, True, True])

这就是我真正想要的：

c = np.array([False, False, 3, 4])
  = np.array([0, 0, 3, 4])

我尝试了各种切片方法，包括：

a[b]
np.extract()
np.choose()
np.take

然而，这些要么不返回数组的形状，要么只返回一个布尔值。

感谢您的帮助。 烤羊肉

Answer 1

你可以使用np.where:

np.where(b, a, 0)
# np.array([0, 0, 3, 4])