关于python:Python代码阅读第44篇寻找符合条件的元素的位置

Python 代码浏览合集介绍：为什么不举荐 Python 初学者间接看我的项目源码

本篇浏览的代码实现了在列表中寻找符合条件的元素的地位。

本篇浏览的代码片段来自于 30-seconds-of-python。

find_index

def find_index(lst, fn):
  return next(i for i, x in enumerate(lst) if fn(x))

# EXAMPLES
find_index([1, 2, 3, 4], lambda n: n % 2 == 1) # 0

find_index函数接管一个列表和一个判别函数，返回第一个使判别函数为真的元素的下标。

函数应用列表推导式、enumerate()和 next() 来返回 lst 中第一个 是判别函数 fn 的返回为 True 的元素的索引。

next(iterator[, default])获取迭代器的下一个元素。如果迭代器耗尽，则返回给定的 default，如果没有默认值则触发 StopIteration

enumerate(iterable, start=0)
返回一个枚举对象。enumerate()返回的迭代器的 __next__() 办法返回一个元组，外面蕴含一个计数值（从 start 开始，默认为 0）和通过迭代iterable 取得的值。如果 start 从0开始，相当于获取了列表的下标。

>>> seasons = ['Spring', 'Summer', 'Fall', 'Winter']
>>> list(enumerate(seasons))
[(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]
>>> list(enumerate(seasons, start=1))
[(1, 'Spring'), (2, 'Summer'), (3, 'Fall'), (4, 'Winter')]

find_last_index

相似的，能够实现 find_last_index 函数，应用 lst[::-1] 切片模式，将列表倒序。而后应用 len(lst) 和倒序后的列表中第一个合乎判别函数的元素的地位，计算该元素在原列表中的地位。

def find_last_index(lst, fn):
  return len(lst) - 1 - next(i for i, x in enumerate(lst[::-1]) if fn(x))

# EXAMPLES
find_last_index([1, 2, 3, 4], lambda n: n % 2 == 1) # 2