python - 「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

マスクされた配列を処理するには、単に「numpy.ma.where」を使用する必要があります。

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