python - 屏蔽 Numpy 数组中的操作

Question

我有两个屏蔽的 numpy 数组。这些是图像。

我正在尝试从另一个中减去一个。

如果我执行标准减法运算符，

ma1 - ma2 

它将减去它们，就好像它们没有被屏蔽一样(不考虑它们的屏蔽)。我希望他们用他们的面具减去。

有谁知道如何从彼此中减去屏蔽的 numpy 数组？

Answer 1

它应该可以工作。当您对掩码数组进行操作时，它采用操作中涉及的掩码的并集。下面的案例显示了 numpy 如何选择在两个掩码数组之间进行减法时将更改的值:

a1 = np.random.random((100,100))
a2 = np.random.random((100,100))

a1 = np.ma.array(a1, mask=a1<0.5)
a2 = np.ma.array(a2, mask=a2<0.5)

umask = np.logical_or(a1.mask, a2.mask) # <-- union of the masks

test = a1.data - a2.data
test[umask] = a1.data[umask] # <-- "canceling" the operation according to the
                             #     combined mask

np.allclose((a1-a2), test)
#True

如你所见，结果是一样的......