我反对将生成器视为列表的诱惑。 简单但幼稚的方法是简单的一线:
gen = (i for i in range(10))
list(gen)[3]
但是请记住,生成器不像列表。 他们不会将中间结果存储在任何地方,因此您不能倒退。 我将在python repl中用一个简单的例子演示这个问题:
>>> gen = (i for i in range(10))
>>> list(gen)[3]
3
>>> list(gen)[3]
Traceback (most recent call last):
File "", line 1, in
IndexError: list index out of range
一旦您开始通过生成器获取序列中的第n个值,生成器现在处于不同的状态,再次尝试获取第n个值将返回不同的结果,这很可能会导致您的错误 码。
让我们看看另一个示例,它基于问题中的代码。
最初,人们会期望以下内容两次打印list(gen)。
gen = (i for i in range(10))
index = 4
for i, v in enumerate(gen):
if i == index:
answer = v
break
print(answer)
for i, v in enumerate(gen):
if i == index:
answer = v
break
print(answer)
但将其键入到repl中,您将得到:
>>> gen = (i for i in range(10))
>>> index = 4
>>> for i, v in enumerate(gen):
... if i == index:
... answer = v
... break
...
>>> print(answer)
4
>>> for i, v in enumerate(gen):
... if i == index:
... answer = v
... break
...
>>> print(answer)
9
祝您早日找到该错误。
编辑:
如前所述,如果生成器无限长,则您甚至无法将其转换为列表。 表达式list(gen)将永远不会完成。
您可以通过一种方法将延迟评估的缓存包装器放在无限生成器周围,使它看起来像可以随意索引的无限长列表,但这值得其问与答,并且会对性能产生重大影响。