Python代码阅读(第46篇)(寻找符合条件的元素)

Python 代码阅读合集介绍: 为什么不推荐Python初学者直接看项目源码
本篇阅读的代码实现了在列表中寻找符合条件的元素。
本篇阅读的代码片段来自于30-seconds-of-python。
find
def find(lst, fn): return next(x for x in lst if fn(x))# EXAMPLES find([1, 2, 3, 4], lambda n: n % 2 == 1) # 1

find函数接收一个列表和一个判别函数,返回第一个使判别函数为真的元素。
函数使用列表推导式和next()来返回lst中第一个 是判别函数fn的返回为True的元素。
【Python代码阅读(第46篇)(寻找符合条件的元素)】next(iterator[, default])获取迭代器的下一个元素。如果迭代器耗尽,则返回给定的 default,如果没有默认值则触发 StopIteration
find_last 类似的,可以实现find_last函数,使用lst[::-1]切片形式,将列表倒序。然后使用相同的方法获取使判别函数fnTrue的元素。
def find_last(lst, fn): return next(x for x in lst[::-1] if fn(x))# EXAMPLES find_last([1, 2, 3, 4], lambda n: n % 2 == 1) # 3

>>> lst = [1,2,3,4,5,6] >>> lst[::-1] [6, 5, 4, 3, 2, 1]

    推荐阅读