掩模numpy阵列并提取符合条件的值

Question

如何正确掩盖两个Numpy阵列？我想找到不等于255的pe值。我还希望我所需的输出数组的大小与pd和pe，ie，（7，7） ，并用填充0是。
实现这一目标的最有效方法是什么？

import numpy as np

pd = np.random.randint(254,256, size=(7,7))
pe = np.random.randint(0,7, size=(7,7))

所需的输出

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

非常感谢

Answer 1

逻辑索引似乎是所有选项中最简单的。

import numpy as np

pd = np.random.randint(254,256, size=(7,7))
pe = np.random.randint(0,7, size=(7,7))

pe[pd == 255] = 0

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

根据您的数据大小，您可以尝试其他选项：

pe = np.where(pd == 255, 0, pe)
# OR
pe = pe * (pd == 255)

但是我猜索引仍然很简单快捷。