我想从匹配条件的列表中获得第一项。产生的方法不能处理整个列表，这一点很重要，因为列表可能相当大。例如，以下函数就足够了:
def first(the_iterable, condition = lambda x: True):
for i in the_iterable:
if condition(i):
return i
这个函数可以这样使用:
>>> first(range(10))
0
>>> first(range(10), lambda i: i > 3)
4
但是，我想不出一个好的内置/单行程序来让我这样做。如果没有必要，我不想复制这个函数。是否有一种内置的方法来获取匹配条件的第一项?
当前回答
下面是三种方法的速度测试。Next()不是最快的方法。
from timeit import default_timer as timer
# Is set irreflexive?
def a():
return frozenset((x3, x3) for x3 in set([x1[x2] for x2 in range(2) for x1 in value]) if (x3, x3) in value) == frozenset()
def b():
return next((False for x1 in value if (x1[0], x1[0]) in value or (x1[1], x1[1]) in value), True)
def c():
for x1 in value:
if (x1[0], x1[0]) in value or (x1[1], x1[1]) in value:
return False
return True
times = 1000000
value = frozenset({(1, 3), (2, 1)})
start_time = timer()
for x in range(times):
a()
print("a(): Calculation ended after " + str(round((timer() - start_time) * 1000) / 1000.0) + " sec")
start_time = timer()
for x in range(times):
b()
print("b(): Calculation ended after " + str(round((timer() - start_time) * 1000) / 1000.0) + " sec")
start_time = timer()
for x in range(times):
c()
print("c(): Calculation ended after " + str(round((timer() - start_time) * 1000) / 1000.0) + " sec")
结果:
Calculation ended after 1.365 sec
Calculation ended after 0.685 sec
Calculation ended after 0.493 sec
其他回答
我知道已经太迟了，但我的回答是:
def find_index(nums, fn):
return next(i for i, x in enumerate(nums) if fn(x))
print(find_index([1, 2, 3, 4], lambda n: n % 2 == 1))
Oneliner:
thefirst = [i for i in range(10) if i > 3][0]
如果你不确定根据条件，任何元素都是有效的，你应该用try/except将其括起来，因为[0]可能引发IndexError。
作为一个可重用、文档化和测试的函数
def first(iterable, condition = lambda x: True):
"""
Returns the first item in the `iterable` that
satisfies the `condition`.
If the condition is not given, returns the first item of
the iterable.
Raises `StopIteration` if no item satysfing the condition is found.
>>> first( (1,2,3), condition=lambda x: x % 2 == 0)
2
>>> first(range(3, 100))
3
>>> first( () )
Traceback (most recent call last):
...
StopIteration
"""
return next(x for x in iterable if condition(x))
带有默认参数的版本
@zorf建议这个函数的一个版本，如果可迭代对象为空或没有匹配条件的项，你可以有一个预定义的返回值:
def first(iterable, default = None, condition = lambda x: True):
"""
Returns the first item in the `iterable` that
satisfies the `condition`.
If the condition is not given, returns the first item of
the iterable.
If the `default` argument is given and the iterable is empty,
or if it has no items matching the condition, the `default` argument
is returned if it matches the condition.
The `default` argument being None is the same as it not being given.
Raises `StopIteration` if no item satisfying the condition is found
and default is not given or doesn't satisfy the condition.
>>> first( (1,2,3), condition=lambda x: x % 2 == 0)
2
>>> first(range(3, 100))
3
>>> first( () )
Traceback (most recent call last):
...
StopIteration
>>> first([], default=1)
1
>>> first([], default=1, condition=lambda x: x % 2 == 0)
Traceback (most recent call last):
...
StopIteration
>>> first([1,3,5], default=1, condition=lambda x: x % 2 == 0)
Traceback (most recent call last):
...
StopIteration
"""
try:
return next(x for x in iterable if condition(x))
except StopIteration:
if default is not None and condition(default):
return default
else:
raise
Python 3中最有效的方法是以下方法之一(使用类似的示例):
“理解”风格:
next(i for i in range(100000000) if i == 1000)
警告:该表达式也适用于Python 2，但在本例中使用的range在Python 3中返回一个可迭代对象，而不是像Python 2那样返回一个列表(如果你想在Python 2中构造一个可迭代对象，则使用xrange代替)。
注意，表达式避免在理解表达式next([i for…])中构造一个列表，这将导致在筛选元素之前创建一个包含所有元素的列表，并将导致处理整个选项，而不是在i == 1000时停止迭代。
具有“功能性”风格:
next(filter(lambda i: i == 1000, range(100000000)))
警告:这在Python 2中不起作用，甚至用xrange替换range，因为过滤器创建了一个列表而不是迭代器(效率低)，下一个函数只适用于迭代器。
默认值
正如在其他响应中提到的那样，如果希望避免在条件不满足时引发异常，则必须向接下来的函数添加一个额外参数。
“功能性”风格:
next(filter(lambda i: i == 1000, range(100000000)), False)
“理解”风格:
使用这种风格，你需要用()包围理解表达式，以避免SyntaxError:如果不是唯一的参数，生成器表达式必须用圆括号括起来:
next((i for i in range(100000000) if i == 1000), False)
对于不存在下一个内置的旧版本的Python:
(x for x in range(10) if x > 3).next()
