如何用NumPy的logical_and对布尔子数组取True值交集
解决方案
直接使用np.logical_and的reduce方法,指定沿第0轴(即最外层维度)做归约运算即可,无需手动拆分外层子数组。
可运行代码示例
先构造你给出的三维测试数组:
import numpy as np arr = np.array([ [[True, True], [True, False]], [[True, False], [False, True]] ])
沿最外层轴执行逻辑与归约:
result = np.logical_and.reduce(arr, axis=0) print(result)
运行后输出和你预期完全一致:
[[ True False] [False False]]
不同写法的适用场景
- 上述
reduce写法是通用方案:无论最外层维度有多少个子数组,都会逐位置判断是否所有子数组在该位置均为True,不需要修改代码逻辑。 - 如果你明确知道最外层固定只有2个子数组,有两种更简洁的写法,结果和上面完全一致:
这两种写法代码更短,但如果后续最外层子数组数量增加,就需要手动调整传参,灵活性不如reduce方案。# 写法1:直接传入两个子数组调用logical_and result = np.logical_and(arr[0], arr[1]) # 写法2:用布尔数组的按位与运算符 result = arr[0] & arr[1]
内容的提问来源于stack exchange,提问作者Bob
相关产品推荐
相关产品推荐

