python - 根据内容过滤字符串列表

Question

给定列表 ['a','ab','abc','bac']，我想计算一个包含 'ab' 字符串的列表> 在其中。 IE。结果是 ['ab','abc']。这如何在 Python 中完成？

Answer 1

这种简单的过滤可以通过 Python 以多种方式实现。最好的方法是使用“列表推导”，如下所示:

>>> lst = ['a', 'ab', 'abc', 'bac']
>>> [k for k in lst if 'ab' in k]
['ab', 'abc']

另一种方法是使用 filter 函数。在 Python 2 中:

>>> filter(lambda k: 'ab' in k, lst)
['ab', 'abc']

在 Python 3 中，它返回 iterator而不是一个列表，但你可以转换它:

>>> list(filter(lambda k: 'ab' in k, lst))
['ab', 'abc']

虽然使用理解是更好的做法。