百木园-与人分享,
就是让自己快乐。

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

来源:https://www.cnblogs.com/felixz/p/15661420.html
图文来源于网络,如有侵权请联系删除。

未经允许不得转载:百木园 » Python代码阅读(第44篇):寻找符合条件的元素的位置

相关推荐

  • 暂无文章