在masked_array上调用numpy.where

Question

Here's an example:

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

Expected output:

[5, 6]

Actual output:

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

How can I achieve the expected output? Thanks! :)

Answer 1

您只需要使用“ numpy.ma.where”即可处理屏蔽的数组：

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