一尘不染

Python-从符合条件的可迭代项中获取第一项

python

我想从符合条件的列表中获得第一项。重要的是,生成的方法不能处理整个列表,这可能会很大。例如,以下功能是足够的:

def first(the_iterable, condition = lambda x: True):
    for i in the_iterable:
        if condition(i):
            return i

可以使用以下功能:

>>> first(range(10))
0
>>> first(range(10), lambda i: i > 3)
4

但是,我想不出一个好的内置式/单层式来让我做到这一点。如果不需要,我特别不想复制此功能。是否有内置的方法来获取与条件匹配的第一项?


阅读 817

收藏
2020-02-15

共1个答案

一尘不染

在Python 2.6或更高版本中:

如果StopIteration在找不到匹配元素的情况下希望被引发:

next(x for x in the_iterable if x > 3)

如果你希望返回default_value(例如None),请执行以下操作:

next( (x for x in the_iterable if x>3), default_value)

请注意,在这种情况下,你需要在生成器表达式周围加一对括号-当生成器表达式不是唯一的参数时,总是需要它们。

我看到大多数答案都坚决地忽略了next内置函数,因此我认为出于某种神秘的原因,它们100%专注于2.5版及更早的版本-并未提及Python版本问题(但后来我没有看到该提及确实提到next内置答案的原因,这就是为什么我认为有必要自己提供答案的原因-至少以这种方式记录“正确版本”问题;-)。

在2.5中,如果迭代器立即完成.next(),则迭代器的方法立即提高StopIteration-即,对于你的用例,如果可迭代项中没有项满足条件。如果你不在乎(即,你知道必须至少有一个令人满意的项目),则只需使用.next()(在genexp上最好next,Python 2.6内置的行及更高版本)。

如果你确实愿意的话,按照你在Q中首先指出的方法将内容包装在函数中似乎是最好的,虽然你建议的函数实现很好,但是你也可以使用itertools,for...: break循环或genexp,或者将a try/except StopIteration作为函数的主体,如各种答案所示。这些替代方案都没有太多附加值,因此我会选择你最初提出的简洁的版本。

2020-02-15