问题描述
我有一个包含 n 行的布尔掩码的 2D numpy 数组,其中每行是一个包含 m 个掩码的数组。
maskArr = [
[[True,False,True,False],[True,True],True]],[[False,[False,[[True,True]]
]
有没有办法对每一行中的掩码数组组合进行向量化,得到如下结果?
combinedMaskArr = [
[True,True]
]
提前感谢您的任何指导或建议。
解决方法
您正在尝试测试沿特定轴的所有元素是否都为真。使用np.all
np.all(maskArr,axis=1)
输出
array([[ True,False,False],[False,True,True],[ True,True]])