Python：创建生成器的两种方法

最新推荐文章于 2025-03-11 16:03:01 发布

ETalien_

最新推荐文章于 2025-03-11 16:03:01 发布

阅读量1k

点赞数

分类专栏： Python 文章标签： python 生成器

本文链接：https://blog.youkuaiyun.com/ETalien_/article/details/107280941

版权

Python 专栏收录该内容

19 篇文章

订阅专栏

本文探讨了Python生成器的原理和用途，它允许一边循环一边计算。介绍了两种创建生成器的方法，并强调了生成器函数在执行流程上的特殊性——每次通过next()调用时从上次yield语句处继续执行。此外，文章还指出，当for循环使用生成器时，无法直接获取return语句的返回值，需通过捕获StopIteration异常来获取。

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

生成器：（一边循环一边计算的机制）

生成器不但可以作用于for循环，还可以被next()函数不断调用并返回下一个值，直到最后抛出StopIteration错误表示无法继续返回下一个值了
第一种创建生成器的方法

g = (x * x for x in range(10))
for i in g:
    print(i)

第二种创建生成器的方法

# 斐波那契正常写法
def fib(num):
    n, a, b = 0, 0, 1
    while n < num:
        print(b)
        a, b = b, a+b
        n = n + 1
    return ‘done’

# 生成器的写法
def fib(num):
    n, a, b = 0, 0, 1
    while n < num:
        yield b
        a, b = b, a+b
        n = n + 1
    return ‘done'

# 一般直接用for循环迭代
for i in fib(6):
    print(i)

如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator。
generator和函数的执行流程不一样。函数是顺序执行，遇到return语句或者最后一行函数语句就返回。而变成generator的函数，在每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处继续执行。

调用generator时，首先要生成一个generator对象，然后用next()函数不断获得下一个返回值，例如

def odd():
    print('step 1')
    yield 1
    print('step 2')
    yield 2
    print('step 3')
    yield 3

o = odd()
print(next(o))  # step 1 
print(next(o))  # step 2 
print(next(o))  # step 3

但是用for循环调用generator时，发现拿不到generator的return语句的返回值。如果想要拿到返回值，必须捕获StopIteration错误，返回值包含在StopIteration的value中

g = fib(6)
while True:
    try:
        x = next(g)
        print('g:', x)
    except StopIteration as e:
        print('Generator return value:', e.value)
        break