如何测试numpy数组中的每个元素是否都被屏蔽

Question

如何测试numpy数组中的每个元素是否都被屏蔽了？这是我想要做的：

x = #is a maksed numpy array
masked_min = numpy.ma.min(x)
if masked_min IS NOT A MASKED ELEMENT:
  #do some stuff only if masked_min is a value

在实践中，我看到这样的情况：

>>> x = numpy.ma.array(numpy.array([1,2,3]),mask=[True,True,True])
>>> masked_min = numpy.ma.min(x)
masked

测试masked没有任何帮助：

>>> numpy.ma.sum(x) == numpy.ma.masked
masked

Answer 1

我认为最有Pythonic风格的方法是只做你想做的事情，然后在整个数组被屏蔽的情况下捕获任何错误。

例如：

x = numpy.ma.array(numpy.array([1,2,3]),mask=[True,True,True])
try: 
    my_min = numpy.min(x[~x.mask])
    # Continue with my_min
except ValueError: 
    # Bail or what have you
    print 'Masks all around!'

Answer 2

如果您所说的“测试是否在numpy中屏蔽了整个数组”是指是否每个元素都被屏蔽了，因为掩码本身是一个数组，那么可以使用.mask.all()

>>> x = numpy.ma.array(numpy.array([1,2,3]),mask=[True,True,True])
>>> x
masked_array(data = [-- -- --],
             mask = [ True  True  True],
       fill_value = 999999)

>>> x.mask
array([ True,  True,  True], dtype=bool)
>>> x.mask.all()
True

或者是.count()，但这是依赖于轴的。OTOH如果你真的只需要测试min调用的结果是否被“屏蔽”了，你可以直接这样做：

>>> numpy.ma.min(x)
masked
>>> type(_)
<class 'numpy.ma.core.MaskedConstant'>
>>> isinstance(numpy.ma.min(x), numpy.ma.core.MaskedConstant)
True
>>> numpy.ma.min(x) is numpy.ma.masked
True

可能有更简单的方法可以做到这一点；我自己很少使用掩码数组。