python更改数组中的部分值_Python：如何根据条件替换掩码数组中的值

我有一个阵列.我想替换值> 5为1,值为< = 5为0.

我还必须考虑无效值(999).

1)我的阵列：

>>> import numpy

>>> a = numpy.array([ [[2, 5, 999],[0, 12, 1]], [[999, 8, 7],[7, 11, 6]] ])

>>> a

array([[[ 2, 5, 999],

[ 0, 12, 1]],

[[999, 8, 7],

[ 7, 11, 6]]])

2)我屏蔽了无效值：

>>> mask_a = (a==999)

>>> a_masked = numpy.ma.masked_array(a, mask = mask_a)

>>> print a_masked

[[[2 5 --]

[0 12 1]]

[[-- 8 7]

[7 11 6]]]

3)我用零替换值< = 5：

>>> a_masked[a_masked<=5]=0

>>> print a_masked

[[[0 0 --]

[0 12 0]]

[[-- 8 7]

[7 11 6]]]

4)我想现在替换值> 5与1：

>>> a_masked[a_masked>5]=1

>>> print a_masked

[[[0 0 1]

[0 1 0]]

[[1 1 1]

[1 1 1]]]

为什么不考虑已被掩盖的值= 999？

我想得到以下结果：

[[[0 0 --]

[0 1 0]]

[[-- 1 1]

[1 1 1]]]