乐趣区

关于python: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 的元素。

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]
退出移动版