優化調用來檢索掩碼數組的掩碼元素的索引？

Question

我有一個屏蔽數組：

a = np.arange(7) 
a = np.ma.masked_greater(a,4) 

一則包含

masked_array(data = [0 1 2 3 4 -- --], 
      mask = [False False False False False True True], 
     fill_value = 999999) 

什麼我要找的是現在檢索，列出每個蒙面元素，即該指數的陣列的有效方式

res = [5, 6] 

不通過掩模像這樣循環：

res = [] 
for idx, data in enumerate(np.ma.getmaskarray(a)): 
    if data: 
     res.append(idx) 

Answer 1

>>> a 
masked_array(data = [0 1 2 3 4 -- --], 
      mask = [False False False False False True True], 
     fill_value = 999999) 
>>> np.where(np.ma.getmaskarray(a)) 
(array([5, 6]),)