python - 查找在 Numpy 中被屏蔽的位置

Question

我想在蒙面数组中获得位置。像这样

wt[chl > 10] = 3
wt[(chl < 10) & (chl > 5)] = 2
wt[(chl < 5)] = 1
wt[(chl is masked )]=0

wt 和 chl 形状相同。我想根据 chl 的值和属性（是否屏蔽）给 wt 值。

有谁知道我应该怎么做？

先感谢您。

Answer 1

argwhere返回值非零的索引，因此如果形状相等，

wt[np.argwhere(chi)] = 0

应该在非零的wt地方归零chi（并且在为零~np.argwhere(chi)的地方归零索引）。chi

Answer 2

您可以直接访问 的mask属性chl并将其用作索引：

wt[chl.mask] = 0