Python 代码阅读（第 44 篇）：寻找符合条件的元素的位置

Felix

关注

发布于: 刚刚

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))
# EXAMPLESfind_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))
# EXAMPLESfind_last_index([1, 2, 3, 4], lambda n: n % 2 == 1) # 2

复制代码

发布于: 刚刚阅读数: 2

原文链接:【http://xie.infoq.cn/article/c56718ced89842f7f86d17865】。

Felix

关注

没有看点，仅仅闲聊。欢迎关注“没看点” 2018.05.04 加入

欢迎关注个人公众号：没看点个人主页：www.zhangfelix.com

发布

暂无评论

创作场景

Python 代码阅读（第 44 篇）：寻找符合条件的元素的位置

find_index

find_last_index

Felix

评论

`find_index`

`find_last_index`