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

Question

我正在做一个图像分割，我在其中创建了一个 mask (bool 数组)。我现在想在保持图像阵列形状的同时提取掩模区域的成像数据。

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

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

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

感谢您的帮助。烧烤栎

Answer 1

你可以使用np.where:

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