numpy数组中的掩蔽值

Question

我有福勒。numpy中的代码：

mask_cntr = np.copy(map_ccodes)
mask_cntr[mask_cntr == cntr] = 1.0
mask_cntr[mask_cntr != 1.0] = 0.0

这里，我将2D数组map_ccodes复制到mask_cntr，并将该数组中等于cntr的值赋给1.0，将所有其他值赋给0.0。你知道吗

在numpy有没有一种更快更像Python的方法？你知道吗

Answer 1

试试看

mask_cntr = 1.0*(map_ccodes==cntr)

我假设cntr == 1来自您的代码？你知道吗

为什么你需要一个单独的面具呢？你可以在任何地方使用map_ccodes==cntr参数。。。你知道吗

Answer 2

np.where函数接受条件并根据条件为真或假返回输出：

np.where(mask_cntr == cntr, 1.0, 0.0)