通过谓词过滤 Python 列表

问题描述：

我想要做类似的事情：

>>> lst = [1, 2, 3, 4, 5]
>>> lst.find(lambda x: x % 2 == 0)
2
>>> lst.findall(lambda x: x % 2 == 0)
[2, 4]

Python 的标准库中是否存在类似这种行为的内容？

我知道在这里自己动手很容易，但我正在寻找一种更标准的方法。

解决方案 1：

您可以使用过滤方法：

>>> lst = [1, 2, 3, 4, 5]
>>> filter(lambda x: x % 2 == 0, lst)
[2, 4]

或者列表推导：

>>> lst = [1, 2, 3, 4, 5]
>>> [x for x in lst if x %2 == 0]
[2, 4]

要查找单个元素，您可以尝试：

>>> next(x for x in lst if x % 2 == 0)
2

但是如果没有匹配项，则会引发异常，因此您可能希望将其包装在 try/catch 中。 () 括号使其成为生成器表达式，而不是列表推导。

不过就我个人而言，我只会使用常规过滤器/理解并取第一个元素（如果有的话）。

如果没有找到任何内容，则会引发异常

filter(lambda x: x % 2 == 0, lst)[0]
[x for x in lst if x %2 == 0][0]

这些返回空列表

filter(lambda x: x % 2 == 0, lst)[:1]
[x for x in lst if x %2 == 0][:1]

解决方案 2：

生成器和列表推导比可链接函数更具 Python 风格。

>>> lst = [i for i in range(1, 6)]

>>> lst
[1, 2, 3, 4, 5]

>>> gen = (x for x in lst if x % 10 == 0)

>>> next(gen, 'not_found')
'not_found'

>>> [x for x in gen]
[]

例如，我有时这样使用它：

>>> n = next((x for x in lst if x % 10 == 0), None)
>>> if n is None:
...     print('Not found')
... 
Not found

否则，您可以像这样定义实用程序函数一行：

>>> find = lambda fun, lst: next((x for x in lst if fun(x)), None)
>>> find(lambda x: x % 10 == 0, lst)
>>> find(lambda x: x % 5 == 0, lst)
5

>>> findall = lambda fun, lst: [x for x in lst if fun(x)]
>>> findall(lambda x: x % 5 == 0, lst)
[5]