Python numpy中的掩码数组操作

这是我在stackoverflow上的第一篇帖子，请耐心等待。 当然，我试图在网上找到答案，但没有成功

问题是：

In [1]: import numpy
In [2]: import numpy.ma as ma
In [4]: a = ma.array([[[1,2],[3,4]],[[1,2],[3,4]]], mask=[[[True,False],[False,False]],[[False,False],[False,True]]])
In [5]: a
Out[5]: 
masked_array(data =
 [[[-- 2]
  [3 4]]

 [[1 2]
  [3 --]]],
             mask =
 [[[ True False]
  [False False]]

 [[False False]
  [False  True]]],
       fill_value = 999999)

In [6]: ma.mean(a, axis=0)
Out[6]: 
masked_array(data =
 [[1.0 2.0]
 [3.0 4.0]],
             mask =
 [[False False]
 [False False]],
       fill_value = 1e+20)

但我希望均值函数返回屏蔽输出，如中所示

In [7]: (a[0]+a[1])/2
Out[7]: 
masked_array(data =
 [[-- 2]
 [3 --]],
             mask =
 [[ True False]
 [False  True]],
       fill_value = 999999)

我哪里做错了？

掩码数组忽略掩码值，它们不会传播掩码。要获得您想要的结果，您可以执行以下操作：

>>> np.ma.array(a.data.mean(axis=0), mask=a.mask.any(axis=0))
masked_array(data =
 [[-- 2.0]
 [3.0 --]],
             mask =
 [[ True False]
 [False  True]],
       fill_value = 1e+20)

---

谢谢@behzad.nouri我喜欢

.mask.any（）

方法，因为它可以在任何一系列数学运算之后应用。