在`masked_array`上调用`numpy.where`

Question

以下是一个例子：

import numpy
from numpy import arange, where
from numpy.ma import masked_array

a = masked_array(arange(10), arange(10) < 5)
print(where((a <= 6))[0])

预期产出：

[5, 6]

实际输出：

[0, 1, 2, 3, 4, 5, 6]

如何实现预期的输出？谢谢！ ：）

Answer 1

你只需要使用＆＃34; numpy.ma.where＆＃34;为了处理蒙面数组：

print(numpy.ma.where((a <= 6))[0])