Aufruf `numpy.where` auf ein` masked_array`

Question

Hier ist ein Beispiel:

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

Erwartete Ausgabe:

[5, 6] 

tatsächliche Ausgang:

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

Wie kann ich erreichen, die erwartete Ausgabe? Vielen Dank! :)

Answer 1

Sie müssen einfach nur "numpy.ma.where" verwenden, um das maskierte Array zu behandeln:

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