python - 基于 bool np.array 有效地对 np.matrix 进行子集化，但仅直到某个阈值

Question

假设我们有一个矩阵 M

M=np.array([[1,2,3],
            [1,2,3],
            [1,2,3],
            [1,2,3]
            [1,2,3],
            [1,2,3]])

行数与 np.array 掩码的长度相同:

mask = np.array([False,True,False,True,False,True])

还有一个名为threshold=2的附加参数

我只想用掩码对 M 进行子集化，直到 2 个 True 值为止，对于所有剩余的值，我将设置为 false。这意味着 M[mask] 应该只返回第二行和第四行，而不是最后一行。有没有一种有效的方法可以通过 numpy 避免 for 循环来做到这一点？

Answer 1

M[掩码][:2]

这会根据掩码选择行，并在达到 2 个 True 值后停止。