Python中的生成器和for循環

Question

所以我有一個生成器函數，看起來像這樣。

def generator():
    while True:
        for x in range(3):
            for j in range(5):
                yield x

在加載此函數並多次調用“ next”之后，我希望它會產生值

0 0 0 0 0 1 1 1 1 1 2 2 2 2 2 0 0 0 0 0 ...

但是相反，它總是一直產生0。 這是為什么？

>>> execfile("test.py")
>>> generator
<function generator at 0x10b6121b8>
>>> generator().next()
0
>>> generator().next()
0
>>> generator().next()
0
>>> generator().next()
0
>>> generator().next()
0
>>> generator().next()
0
>>> generator().next()
0

Answer 1

generator()初始化新的生成器對象：

In [4]: generator() is generator() # Creating 2 separate objects
Out[4]: False

然后generator().next()從新創建的生成器對象（在您的情況下為0 generator().next()獲取第一個值。

您應該調用一次generator ：

In [5]: gen = generator() # Storing new generator object, will reuse it

In [6]: [gen.next() for _ in range(6)] # Get first 6 values for demonstration purposes
Out[6]: [0, 0, 0, 0, 0, 1]

注意 ： generator.next已從Python 3（ PEP 3114 ）中刪除-使用next函數代替：

In [7]: next(gen)
Out[7]: 1

Answer 2

每次調用generator您都將創建一個新的generator對象：

generator().next() # 1st item in 1st generator 
generator().next() # 1st item in 2nd generator

創建一個生成器，然后調用next一個生成器：

g = generator()

g.next() # 1st item in 1st generator
g.next() # 2nd item in 1st generator

Answer 3

我在評論中發布了它，我想如果您與此問題同時查看，就會發現錯誤：

with open('my_file.txt', 'r') as f: 
    print(f.readline())               # `readline` simply reads a single line

with open('my_file.txt', 'r') as f: 
    print(f.readline())

with open('my_file.txt', 'r') as f: 
    print(f.readline())

with open('my_file.txt', 'r') as f: 
    print(f.readline())    

不必每次都創建一個新的生成器對象，而必須使其一次然后使用它，直到耗盡為止：

mygen = generator()
mygen.next()
mygen.next()
...
etc