掩盖numpy中不按排序顺序排列的值

时间:2015-01-20 20:25:00

标签: python numpy

我有一个复杂的计算,我希望结果是一个数组,其中值按排序顺序排列。但是,由于某些关键点的数值误差,某些结果值是错误的。我想掩盖这些价值观。我该怎么做?

这是一个等效函数,但假设值从最高到最低排序,并且外来者总是大于预期值。我想知道是否有更简单,更有效的方法来做到这一点。

def maskoutsiders(a):
    mask = numpy.zeros(len(a))
    lastval = a[0]
    for i in range(1, len(a)):
        if a[i] > lastval:
            mask[i] = 1
        else:
            lastval = a[i]

    return ma.masked_array(a, mask=mask)

1 个答案:

答案 0 :(得分:4)

a应该减少时,您可以使用:

mask = a > np.minimum.accumulate(a)

a应该增加时,你可以使用:

mask = a < np.maximum.accumulate(a)

npnumpy。)

例如,

In [44]: def mymaskoutsiders(a):
   ....:     mask = a > np.minimum.accumulate(a)
   ....:     return ma.masked_array(a, mask=mask)
   ....: 

将结果与此数组进行比较:

In [100]: x
Out[100]: array([ 13. ,  16.5,  15.5,  11.5,   6. ,   9.5,   5.5,   9. ,   5. ,   2.5])

这是你的功能:

In [101]: maskoutsiders(x)
Out[101]: 
masked_array(data = [13.0 -- -- 11.5 6.0 -- 5.5 -- 5.0 2.5],
             mask = [False  True  True False False  True False  True False False],
       fill_value = 1e+20)

这是我的版本:

In [102]: mymaskoutsiders(x)
Out[102]: 
masked_array(data = [13.0 -- -- 11.5 6.0 -- 5.5 -- 5.0 2.5],
             mask = [False  True  True False False  True False  True False False],
       fill_value = 1e+20)