如何选择包含特定值的 NumPy 矩阵行？

Question

我有以下 NumPy 数组：

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

我想要一个数组，其中包含任意列中m至少出现一次 的行，因此：1

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

Answer 1

使用any和布尔索引：

out = m[(m==1).any(axis=1)]

输出：

array([[1, 2, 3],
       [1, 2, 1]])

中间体：

(m==1)

array([[ True, False, False],
       [False, False, False],
       [ True, False,  True]])


(m==1).any(axis=1)

array([ True, False,  True])