我想要一种惯用的方式来找到与谓词匹配的列表中的第一个元素。
当前代码非常丑陋:
[x for x in seq if predicate(x)][0]
我已经考虑过将其更改为:
from itertools import dropwhile dropwhile(lambda x: not predicate(x), seq).next()
但是必须有一些更优雅的方法……如果返回一个None值而不是没有找到匹配项引发异常,那将是一个很好的选择。
None
我知道我可以像这样定义一个函数:
def get_first(predicate, seq): for i in seq: if predicate(i): return i return None
但是,如果已经有内置的插件开始用这样的实用函数填充代码,这是很鸡肋的(人们可能不会注意到它们已经在那里,因此随着时间的推移它们会不断重复出现)。
要查找seq与匹配的序列中的第一个元素predicate:
seq
predicate
next(x for x in seq if predicate(x))
或(itertools.ifilter在Python 2上):
itertools.ifilter
next(filter(predicate, seq))
StopIteration如果没有,它将引发。
StopIteration
None如果没有这样的元素,则返回:
next((x for x in seq if predicate(x)), None)
要么:
next(filter(predicate, seq), None)