python - 使用 bool 索引数组过滤列表

Question

如何在不使用 numpy 的情况下使用 bool 索引数组来过滤列表？

例如:

>>> l = ['a','b','c']
>>> b = [True,False,False]
>>> l[b]

结果应该是:

['a']

我知道 numpy 支持它，但想知道如何在 Python 中解决。

>>> import numpy as np
>>> l = np.array(['a','b','c'])
>>> b = np.array([True,False,False])
>>> l[b]
array(['a'], 
      dtype='|S1')

Answer 1

Python 不支持 bool 索引但是 itertools.compress功能完全符合您的要求。它返回一个迭代器，这意味着您需要使用 list 构造函数返回一个列表。

>>> from itertools import compress
>>> l = ['a', 'b', 'c']
>>> b = [True, False, False]
>>> list(compress(l, b))
['a']