python - 聚合 numpy 屏蔽数组

Question

我正在与 numpy masked arrays 合作并且有一个微不足道的操作我无法弄清楚如何以简单的方式完成它。如果我有两个屏蔽数组，如何将它们聚合到另一个仅包含未屏蔽值的数组中？

In [1]: import numpy as np

In [2]: np.ma.array([1, 2, 3], mask = [0,1,1])
Out[2]:
masked_array(data = [1 -- --],
             mask = [False  True  True],
       fill_value = 999999)

In [3]: np.ma.array([4, 5, 6], mask = [1,1,0])
Out[3]:
masked_array(data = [-- -- 6],
             mask = [ True  True False],
       fill_value = 999999)

如果我想得到:

masked_array(data = [1 -- 6],
             mask = [False  True False],
       fill_value = 999999)

Answer 1

使用 numpy.dstack 堆叠掩码和数组并创建一个新的屏蔽数组，然后您可以使用 numpy.prod 获得所需的输出:

>>> a1 = np.ma.array([1, 2, 3], mask = [0,1,1])
>>> a2 = np.ma.array([7, 8, 9], mask = [1,1,0])
>>> arr = np.ma.array(np.dstack((a1, a2)), mask=np.dstack((a1.mask, a2.mask)))
>>> np.prod(arr[0], axis=1)
masked_array(data = [1 -- 9],
             mask = [False  True False],
       fill_value = 999999)