Python/Numpy:在二维数组中按行组合布尔掩码

问题描述

我有一个包含 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]])