Python 生成器(yield)

Python 生成器


一、生成器(generator)

  • Python 的生成器是一种特殊的程序,可以被用作控制循环的迭代行为。生成器可以暂时挂起函数,并保留函数的局部变量等数据,然后在再次调用它的时候,从上次暂停的位置继续执行下去。
  • 如果所要的序列的元素可以按照某种算法推算出来,可以采用迭代器一边循环一边计算的机制,这样就不必创建完整的序列,从而节省大量的空间。
  • Python 的生成器是迭代器的一种。
  • 协程是基于生成器实现的。

二、基于列表生成式的生成器

1、创建生成器:列表生成式的 [ ] 改成 ( )

# 创建一个生成器,查看其类型
g = (x * x for x in range(5))
print(g)
>>> <generator object <genexpr> at 0x00000173441AB2B0>

2、使用生成器:next()

每次调用next(g),就计算出生成器g的下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出StopIteration的错误

# 获得生成器的下一个值
next(g)
>>> 0
next(g)
>>> 1
......
next(g)
>>> Traceback (most recent call last):
	  File "<stdin>", line 1, in <module>
	StopIteration

# 遍历生成器
for n in g:
    print(n)
>>> 0 1 4 9 16

三、yield 生成器

1、创建生成器:yield

# 创建一个生成器,查看其类型
def f(n):
    i = 0
    while i < n:
        yield i**2
        i += 1
    return 'finish'

g = f(5)
print(g)
>>> <generator object f at 0x000001F73626B2B0>

2、使用生成器:next()

# 遍历生成器
while True:
    try:
        x = next(g)
        print(x)
    except StopIteration as e:
        print('Generator return value:', e.value)
        break
>>> 0 1 4 9 16
	Generator return value: finish

3、生成器传递参数:send()

  • yield 语句执行返回操作后会马上挂起,等待接收值。send() 值时会马上进行赋值,如果没有接收者,会执行代码直至 yield 语句。
  • 启动生成器:.send(None),第一次使用 send(),这时候没有 yield 接收值,执行函数的初始化代码,直到 yield 语句,返回 yield 后面的值(语句的赋值操作没有执行),生成器挂起。
  • 生成器再次传递值:.send(n),生成器会先接收值 n 赋值给 yield 语句的变量,然后执行 yield 语句之后的代码,直至遇到下一个 yield 语句把 yield 后面的值返回给调用者,生成器再次挂起,等待接收值。
# 消费者
def consumer():
    r = ''
    while True:
        n = yield r
        if not n:
            return
        print('[CONSUMER] Consuming %s...' % n)
        r = 'done'

# 生产者
def produce(c):
    c.send(None)
    n = 0
    while n < 5:
        n = n + 1
        print('[PRODUCER] Producing %s...' % n)
        r = c.send(n)
        print('[PRODUCER] Consumer return: %s' % r)
    c.close()

c = consumer()
produce(c)

>>> [PRODUCER] Producing 1...
	[CONSUMER] Consuming 1...
	[PRODUCER] Consumer return: done
	......
	[PRODUCER] Producing 5...
	[CONSUMER] Consuming 5...
	[PRODUCER] Consumer return: done

GOOD LUCK!


  • 3
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值