掩码 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

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