在 numpy 中为 2D 数组查找 2 个掩码之间的剩余掩码

Question

假设我有一个二维数组：

main = np.random.random((300, 200))

我有两个用于这个数组的掩码： 例如，

mask1 = list((np.random.randint((100), size = 50), np.random.randint((200), size = 50)))
mask2 = list((np.random.randint((20), size = 10), np.random.randint((20), size = 10)))

我想替换二维数组中的主要值，例如：

main[mask1]=2
main[mask2]=1

效果很好，但我也想用零替换所有既不是掩码 1 也不是掩码 2 的索引。

我想过这样的事情：

main[~mask1] & main[~mask2] = 0

这让我无处可去，因此感谢任何帮助！

Answer 1

我认为根据您的要求，更好的方法是构造一个与main 形状相同的零填充数组，并使用mask1 和mask2 分配1 和2

main = np.zeros(main.shape)
main[mask1]=2
main[mask2]=1