替换numpy掩码数组中的值

Question

我试图替换numpymasked array中的一些值，但是我的掩码被删除了：

import numpy as np
a = np.ma.array([1, 2, 3, -1, 5], mask=[0, 0, 0, 1, 0])
a[a < 2] = 999

结果是：

^{pr2}$

但我想要的是：

masked_array(data = [999 2 3 -- 5],
mask = [False False False  True False],
fill_value = 999999)

我做错什么了？ 我在Ubuntu13.10上使用了Python2.7和Numpy1.7.1

Answer 1

我认为你没有正确地进行替换，试试这个：

>>> import numpy as np
>>> a = np.ma.array([1, 2, 3, -1, 5], mask=[0, 0, 0, 1, 0])
>>> a.data[a < 2] = 999
>>> a
 masked_array(data = [999 2 3   5],
         mask = [False False False  True False],
   fill_value = 999999)