为什么我的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()，而不是每次循环都调用。