十八、python沉淀之路--生成器

一、生成器

生成器总结
语法上和函数类似生成器函数和常规函数几乎是一样的。他们都是使用def语句进行定义,差别在于生成器使用yield语句返回一个值,而常规函数使用return语句返回一个值。自动实现迭代器协议;对于生成器,python会自动实现迭代器协议,以便应用到迭代背景种,(如for循环,sum函数)。由于生成器自动实现了迭代器协议,所以,我们可以调用它的next方法,并且,在没有值可以返回的时候,生成器自动产生stop iteration异常。

状态挂起生成器使用yield语句返回一个值,yield语句挂起该生成器函数的状态,保留足够的信息,以便之后从他离开的地方继续执行。

1、初识

1 gift_list = ('礼盒%s'%i for i in range(10))   #这就是生成器表达式
2 print(gift_list)
3 print(gift_list.__next__())
4 print(next(gift_list))
1 <generator object <genexpr> at 0x00000294B972DDB0>
2 礼盒0
3 礼盒1

2、

 1 import time
 2 def test():
 3     print('桃树准备开花了')
 4     time.sleep(1)
 5     yield '长了第一个桃子'
 6     time.sleep(1)
 7     print('把第一个桃子吃了')
 8     time.sleep(1)
 9     yield '长了第二个桃子'
10     print('把第二个桃子吃了')
11     yield '长了第三个桃子'
12 
13 res = test()
14 print(res)
15 print(res.__next__())    #这次执行yield就会停在那里。下一次执行next就从上一次yield的地方开始执行。
16 print(next(res))
17 print(res.__next__())
1 <generator object test at 0x00000121469BA308>
2 桃树准备开花了
3 长了第一个桃子
4 把第一个桃子吃了
5 长了第二个桃子
6 把第二个桃子吃了
7 长了第三个桃子

3、卖包子的例子

1 def produce_baozi():
2     for i in range(1,11):
3         print('准备生产包子')
4         yield '一笼包子%i'%i
5         print('开始卖包子')
6 p_baozi = produce_baozi()
7 print(p_baozi.__next__())
8 print(p_baozi.__next__())
1 准备生产包子
2 一笼包子1
3 开始卖包子
4 准备生产包子
5 一笼包子2

4、产鸡蛋的例子

 1 def produce_egg():
 2     ret = []
 3     for i in range(1,11):
 4         ret.append('鸡蛋-%s'%i)
 5     return ret
 6 
 7 p_egg = produce_egg()
 8 print(p_egg)
 9 
10 这种方式 占空间大,效率低
11 
12 
13 
14 def produce_egg():
15     for i in range(1,11):
16         yield 'egg-%s'%i
17 chicken = produce_egg()
18 print(chicken.__next__())
19 print(chicken.__next__())
20 print(chicken.__next__())
21 print(next(chicken))
22 print(next(chicken))
23 
24 
25 for egg in chicken:
26     print(egg)
 1 ['鸡蛋-1', '鸡蛋-2', '鸡蛋-3', '鸡蛋-4', '鸡蛋-5', '鸡蛋-6', '鸡蛋-7', '鸡蛋-8', '鸡蛋-9', '鸡蛋-10']
 2 egg-1
 3 egg-2
 4 egg-3
 5 egg-4
 6 egg-5
 7 egg-6
 8 egg-7
 9 egg-8
10 egg-9
11 egg-10

优点一生成器的好处是延迟计算,一次返回一个结果。也就是说,它不会一次生成所有的结果。这对于大数据量处理,将会非常有用。

列表解释
sum((i for i in range(1000000000))  #内存占用大,机器容易卡死

生成器表达式
sum(i for i in range(1000000000))  #几乎不占内存

优点二生成器还能有效提高代码可读性

这里,至少有两个充分的理由说明,使用生成器比不使用代码更加清晰:
    1、使用生成器以后,代码行数更少。大家要记住,如果想要把代码写的pythonic在保证代码可读性的前提下,代码行数越少越好。
    2、不使用生成器的时候,对于每次结果,我们首先看到的是result append(index),其次,才是index。也就是说,我们每次看到的是一个列表的append的操作      的,知识append的是我们想要的结果。使用生成器时候,直接yield index,少了列表append操作的干扰,我们一眼就能够看出,代码是要返回index。
    
    上面的例子充分说明了,合理使用生成器,能够有效提高代码可读性,只要大家完全接收了生成器的概念,理解了yield语句和return语句一样,业绩是返回一个值。
    那么,就能够理解为什么使用生成器比不适用生成器要好,能够 理解使用生成器真的可以让代码变得清晰易懂。

二、通过生成器实现协程并行运算

例1

 1 def consumer(name):
 2     print('%s开始准备吃包子了'%name)
 3     while True:
 4         baozi = yield 1
 5         print('包子%s被%s吃完了'%(baozi,name))
 6 
 7 def producer(name):
 8     c = consumer('A')
 9     c2 = consumer('B')
10     print(c.__next__())
11     print(c2.__next__())
12     print('大爷要吃包子啦')
13     for i in range(1,11):
14         c.send(i)
15         c2.send(i)
16 producer('隔壁老王')
 1 A开始准备吃包子了
 2 1
 3 B开始准备吃包子了
 4 1
 5 大爷要吃包子啦
 6 包子1被A吃完了
 7 包子1被B吃完了
 8 包子2被A吃完了
 9 包子2被B吃完了
10 包子3被A吃完了
11 包子3被B吃完了
12 包子4被A吃完了
13 包子4被B吃完了
14 包子5被A吃完了
15 包子5被B吃完了
16 包子6被A吃完了
17 包子6被B吃完了
18 包子7被A吃完了
19 包子7被B吃完了
20 包子8被A吃完了
21 包子8被B吃完了
22 包子9被A吃完了
23 包子9被B吃完了
24 包子10被A吃完了
25 包子10被B吃完了

例2

 1 import time
 2 def consumer(name):
 3     print('我是[%s],我准备开始吃包子了' %name)
 4     while True:
 5         baozi=yield
 6         time.sleep(1)
 7         print('%s 很开心的把【%s】吃掉了' %(name,baozi))
 8 
 9 def producer():
10     c1=consumer('隔壁老王')
11     c2=consumer('对面老李')
12     c1.__next__()
13     c2.__next__()
14     for i in range(10):
15         time.sleep(1)
16         c1.send('包子 %s' %i)
17         c2.send('包子 %s' %i)
18 producer()
 1 我是[隔壁老王],我准备开始吃包子了
 2 我是[对面老李],我准备开始吃包子了
 3 隔壁老王 很开心的把【包子 0】吃掉了
 4 对面老李 很开心的把【包子 0】吃掉了
 5 隔壁老王 很开心的把【包子 1】吃掉了
 6 对面老李 很开心的把【包子 1】吃掉了
 7 隔壁老王 很开心的把【包子 2】吃掉了
 8 对面老李 很开心的把【包子 2】吃掉了
 9 隔壁老王 很开心的把【包子 3】吃掉了
10 对面老李 很开心的把【包子 3】吃掉了
11 隔壁老王 很开心的把【包子 4】吃掉了
12 对面老李 很开心的把【包子 4】吃掉了
13 隔壁老王 很开心的把【包子 5】吃掉了
14 对面老李 很开心的把【包子 5】吃掉了
15 隔壁老王 很开心的把【包子 6】吃掉了
16 对面老李 很开心的把【包子 6】吃掉了
17 隔壁老王 很开心的把【包子 7】吃掉了
18 对面老李 很开心的把【包子 7】吃掉了
19 隔壁老王 很开心的把【包子 8】吃掉了
20 对面老李 很开心的把【包子 8】吃掉了
21 隔壁老王 很开心的把【包子 9】吃掉了
22 对面老李 很开心的把【包子 9】吃掉了

例3

 1 import time
 2 def producer():
 3     ret=[]
 4     for i in range(10):
 5         time.sleep(0.1)
 6         ret.append('包子%s' %i)
 7     return ret
 8 
 9 def consumer(res):
10     for index,baozi in enumerate(res):
11         time.sleep(0.1)
12         print('第%s个人,吃了%s' %(index,baozi))
13 
14 res=producer()
15 consumer(res)
 1 第0个人,吃了包子0
 2 第1个人,吃了包子1
 3 第2个人,吃了包子2
 4 第3个人,吃了包子3
 5 第4个人,吃了包子4
 6 第5个人,吃了包子5
 7 第6个人,吃了包子6
 8 第7个人,吃了包子7
 9 第8个人,吃了包子8
10 第9个人,吃了包子9
yield 3相当于return 控制的是函数的返回值
x=yield的另外一个特性,接受send传过来的值,赋值给x

例4

 1 def test():
 2     print('开始啦')
 3     firt=yield #return 1   first=None
 4     print('第一次',firt)
 5     yield 2
 6     print('第二次')
 7 
 8 t=test()
 9 res=t.__next__() #next(t)
10 print(res)
11 # t.__next__()
12 # res=t.send(None)
13 res=t.send('函数停留在first那个位置,我就是给first赋值的')
14 print(res)
1 开始啦
2 None
3 第一次 函数停留在first那个位置,我就是给first赋值的
4 2
原文地址:https://www.cnblogs.com/jianguo221/p/8977465.html