Python生成器

原创转载请注明出处：http://agilestyle.iteye.com/blog/2329232

 

将一个列表生成式的[]改为()，就创建了一个generator

l = [x * x for x in range(1, 6)]
print(l)

g1 = (x * x for x in range(1, 6))
print(g1)
for i in g1:
    print(i)

Console Output

 

generator保存的是算法，每次调用next(g)，就计算出g的下一个元素的值，直到计算到最后一个元素，没有更多的元素时，抛出StopIteration的错误

g2 = (x * x for x in range(1, 6))
print(next(g2))
print(next(g2))
print(next(g2))
print(next(g2))
print(next(g2))
print(next(g2))

Console Output

 

使用generator定义斐波拉契数列

# 如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator
# generator和函数的执行流程不一样。函数是顺序执行，遇到return语句或者最后一行函数语句就返回。
# 而变成generator的函数，在每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处继续执行。
def fib(m):
    n, a, b = 0, 0, 1
    while n < m:
        # print(b)
        # 要把fib函数变成generator，只需要把print(b)改为yield b就可以了
        yield b
        a, b = b, a + b
        n = n + 1
    return 'done'


for i in fib(5):
    print(i)

# 用for循环调用generator时，发现拿不到generator的return语句的返回值
# 如果想要拿到返回值，必须捕获StopIteration错误，返回值包含在StopIteration的value中
g = fib(5)
while True:
    try:
        i = next(g)
        print(i)
    except StopIteration as e:
        print(e.value)
        break

Console Output