我有一个蒙版数组:
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)
>>> 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]),)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句