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

发布于 2021-02-02 23:19:41

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

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

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

关注者
0
被浏览
82
1 个回答
  • 面试哥
    面试哥 2021-02-02
    为面试而生,有面试问题,就找面试哥。

    在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作为函数的主体,如各种答案所示。这些替代方案都没有太多附加值,因此我会选择你最初提出的简洁的版本。



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看