理解Python生成器-优快云博客

本文介绍了Python中的生成器概念，展示了如何通过简单的语法更改从列表推导式转换为生成器表达式，并提供了使用生成器实现斐波那契数列的具体实例。此外还解释了generator的工作原理及其优势。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原创转载请注明出处：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