如何基于numpy数组中的条件删除行?

笔记 numpy 支持矢量化比较:

>>> test
array([[1, 2, 'a'],
       [4, 5, 6],
       [7, 'a', 9],
       [10, 11, 12]], dtype=object)
>>> test == 'a'
array([[False, False,  True],
       [False, False, False],
       [False,  True, False],
       [False, False, False]], dtype=bool)

现在,你想要 排 哪里 全部的 不等于 'a' :

>>> (test != 'a').all(axis=1)
array([False,  True, False,  True], dtype=bool)

因此,只需使用掩码选择行:

>>> row_mask = (test != 'a').all(axis=1)
>>> test[row_mask,:]
array([[4, 5, 6],
       [10, 11, 12]], dtype=object)

还有,也许像这样?(灵感来源于 another answers )

In [100]: mask = ~(test == 'a')

In [101]: mask
Out[101]: 
array([[ True,  True, False],
       [ True,  True,  True],
       [ True, False,  True],
       [ True,  True,  True]], dtype=bool)

In [102]: test[np.all(mask, axis=1), :]
Out[102]: 
array([['4', '5', '6'],
       ['10', '11', '12']],
      dtype='<U21')

但是,请注意,我们在这里 不 正在删除 原始数组中的任何行。我们只是把没有字母表的行切掉 a .

总之,有几种可能的方法,例如:

test[np.all(test != 'a', axis=1), :]

或

test[(test != 'a').all(axis=1)]