forked from yidao620c/python3-cookbook
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathyield_send.py
More file actions
71 lines (62 loc) · 2.71 KB
/
yield_send.py
File metadata and controls
71 lines (62 loc) · 2.71 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
#!/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)