python取 n t中间值_在Python中获取生成器的第n个项

我反对将生成器视为列表的诱惑。 简单但幼稚的方法是简单的一线：

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)将永远不会完成。

您可以通过一种方法将延迟评估的缓存包装器放在无限生成器周围，使它看起来像可以随意索引的无限长列表，但这值得其问与答，并且会对性能产生重大影响。