為什么我的Python測試生成器根本不起作用？

Question

這是一個示例腳本，用於測試yield的使用...我做錯了嗎？ 它總是返回'1'...

#!/usr/bin/python

def testGen():
    for a in [1,2,3,4,5,6,7,8,9,10]:
         yield a

w = 0
while w < 10:
    print testGen().next()
        w += 1

Answer 1

您每次都在創建一個新的生成器。 您只應調用一次testGen() ，然后使用返回的對象。 嘗試：

w = 0
g = testGen()
while w < 10:
    print g.next()
    w += 1

當然，還有正常的慣用生成器用法：

for n in testGen():
    print n

請注意，這只會在循環開始時調用一次testGen() ，而不是每次迭代調用一次。