python 理解生成器的意外行为[重复]

Question

最近我在 python 的理解生成器中遇到了一些奇怪的行为

我的代码：

with open('test.txt') as file:
    lines = (line for line in file.read().splitlines())

for line in lines:
    print(line)
for line in lines:
    print(line)

但是，我只得到一个，而不是对同一文件内容进行两次迭代：

row1
row2
row3
...

我期望的地方：

row1
row2
row3
...

row1
row2
row3
...

我在其他论坛上找不到此问题的解决方案，所以如果您知道答案，我将不胜感激：-)

编辑：这个问题是由于我对生成器的工作原理缺乏了解而发布的。我确信，一旦我完成了一个生成器的循环，我就可以根据需要再做一个又一个。对于这个问题，我深表歉意。

Answer 1

你应该使用列表推导而不是生成器，用方括号替换括号，比如

with open('test.txt') as file:
    lines = [line for line in file.read().splitlines()] # parenthesis replaced with square brackets

for line in lines:
    print(line)
for line in lines:
    print(line)