如何用Numpy实现掩码A减去掩码B的逻辑运算？

你需要的运算逻辑等价于仅当A为1且B为0时结果为1，其余情况均为0，对应逻辑表达式 A ∧ ¬B（A与非B），可以用以下几种简洁方式实现：

针对布尔型掩码数组 #

如果A和B是bool类型的数组，直接用按位逻辑运算符即可：

import numpy as np

# 示例布尔数组
A = np.array([[1, 1, 0, 0]], dtype=bool)
B = np.array([[1, 0, 1, 0]], dtype=bool)

result = A & ~B
# 若需要转为0-1整数数组，可追加.astype(int)
# result = (A & ~B).astype(int)

针对整数型0-1掩码数组 #

如果A和B是int类型的0-1数组，有两种简洁实现：

1. 用算术运算实现逻辑非，再做按位与：

result = A & (1 - B)

1 - B会将B中的0转为1、1转为0，正好对应逻辑非，再与A按位与就得到预期结果。

2. 减法后截断负数：

result = (A - B).clip(min=0)

这种方式利用减法将不符合条件的情况（A=0、B=1）得到的-1截断为0，也能满足需求，但逻辑直观性不如第一种。

为什么直接用A - B不行？ #

当A元素为0、B元素为1时，A - B会得到-1，这不符合0-1掩码的要求，因此不能直接使用减法。

内容的提问来源于stack exchange，提问作者user16971617