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

时间:2012-06-03 04:52:47

标签: python numpy

如何测试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

2 个答案:

答案 0 :(得分:13)

如果通过"测试整个阵列是否被掩盖了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如果你真的只需要测试你的最小通话结果是否被掩盖了#34;你可以直接这样做:

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

[可能有更简单的方法来做到这一点;我自己很少使用蒙面数组。]

答案 1 :(得分:4)

我认为最恐怖的方式就是做你想做的事情,然后捕捉整个数组被掩盖后产生的任何错误。

例如:

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!'