72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
#!/usr/bin/env python
|
||
# -*- encoding: utf-8 -*-
|
||
"""
|
||
Topic: 使用yield和send的典型场景
|
||
Desc :
|
||
get_primes的后几行需要着重解释。yield关键字返回number的值，而像 other = yield foo 这样的语句的意思是，
|
||
"返回foo的值，这个值返回给调用者的同时，将other的值也设置为那个值"。
|
||
你可以通过send方法来将一个值”发送“给生成器，这时候就是将other值设置为发送的值了。
|
||
def get_primes(number):
|
||
while True:
|
||
if is_prime(number):
|
||
number = yield number
|
||
number += 1
|
||
通过这种方式，我们可以在每次执行yield的时候为number设置不同的值。
|
||
现在我们可以补齐print_successive_primes中缺少的那部分代码：
|
||
def print_successive_primes(iterations, base=10):
|
||
prime_generator = get_primes(base)
|
||
prime_generator.send(None)
|
||
for power in range(iterations):
|
||
print(prime_generator.send(base ** power))
|
||
这里有两点需要注意：
|
||
首先，我们打印的是generator.send的结果，这是没问题的，
|
||
因为send在发送数据给生成器的同时还返回生成器通过yield生成的值（就如同生成器中yield语句做的那样）。
|
||
第二点，看一下prime_generator.send(None)这一行，
|
||
当你用send来“启动”一个生成器时（就是从生成器函数的第一行代码执行到第一个yield语句的位置），
|
||
你必须发送None。这不难理解，根据刚才的描述，生成器还没有走到第一个yield语句，
|
||
如果我们发送一个真实的值，这时是没有人去“接收”它的。一旦生成器启动了，我们就可以像上面那样发送数据了。
|
||
"""
|
||
import random
|
||
|
||
|
||
def get_data():
|
||
"""返回0到9之间的3个随机数"""
|
||
return random.sample(range(10), 3)
|
||
|
||
|
||
def consume():
|
||
"""显示每次传入的整数列表的动态平均值"""
|
||
running_sum = 0
|
||
data_items_seen = 0
|
||
|
||
while True:
|
||
print('before 1 yield....')
|
||
data = yield [0, 0, 0]
|
||
print('-------yield inner------- {}'.format(data))
|
||
print('after 1 yield...')
|
||
data_items_seen += len(data)
|
||
running_sum += sum(data)
|
||
print('The running average is {} - {} - {}'.format(
|
||
data_items_seen, running_sum, running_sum / float(data_items_seen)))
|
||
|
||
|
||
def produce(consumer):
|
||
"""产生序列集合，传递给消费函数（consumer）"""
|
||
while True:
|
||
data = get_data()
|
||
print('Produced {}'.format(data))
|
||
consumer.send(data)
|
||
yield
|
||
|
||
|
||
if __name__ == '__main__':
|
||
consumer = consume()
|
||
aa = consumer.send(None)
|
||
print(aa)
|
||
bb = consumer.send(get_data())
|
||
print(bb)
|
||
producer = produce(consumer)
|
||
# for _ in range(2):
|
||
# print('Producing...')
|
||
# next(producer)
|