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!