Python笔记_从迭代器、生成器到协程(一)

最近看了一篇老的教程,结合最近开始使用的python3,简单介绍一下python中协程的用法,本文中实例的代码都是在python3.5中可以运行的。

协程的内容受以下教程的启发较大:

http://dabeaz.com/coroutines/

1、迭代器(Iterator)VS. 生成器(Generator)?

这两个概念相关性很大,简单来说:

  • 迭代器就是实现了__next__和__iter__方法的对象,这个对象可以被迭代
  • 生成器是用函数+yield的方式实现迭代协议的一种语法,它看似函数、但却不是普通的函数,而且生成器不仅能实现迭代、还能实现协程的功能。

先看一个手动定义一个迭代器的例子:

#定义一个迭代器类
from random import randint

class Iter(object):
    def __init__(self,n):
        #初始化时确定迭代器的最大迭代次数
        self.n = n
        self.i= 0
    
    def __iter__(self):
        return self

    def __next__(self):
        if self.i < self.n:
            self.i = self.i+1
            #每次迭代都返回一个100以内的随机数
            return randint(0,100)
        else:
            #迭代器中总需要返回一个StopIteration的异常退出迭代
            raise StopIteration

if __name__ == '__main__':
    #实例化一个迭代器对象
    c=Iter(10)
    for x in c:
    print(x)
    #还有一种迭代方法
    print('\n')
    d=Iter(20)
    print(d.__next__())
    print(d.__next__())
    print(d.__next__())
输出如下:
44
54
65
22
83
30
86
87
75
46


100
9
1

实现的同样的迭代逻辑,来看看生成器的代码的写法:

#定义一个生成器
from random import randint
def Gen(n):
    i=0 
    while (i < n): 
        i=i+1
        yield randint(0,100)

if __name__ == '__main__':
    c=Gen(10)
    for x in c:
        print (x)

从上面的代码可以看出,实现相同的逻辑,定义一个生成器远比写一个迭代器简单。迭代器是写一个类,生成器是定义一个函数。

对比一下上面两个例子的迭代器的对象和生成器函数的返回是不一样的,但是,他们都实现了同一个功能

  1. 一个可迭代的对象<sample1.Iter object at 0x7f855b64aa58>
  2. 一个生成器对象<generator object Gen at 0x7f855b647f68>

2、生成器的yield语法糖

yield是一个很有意思的语法糖,如果在一个函数中有yield,那么这个函数就不再是一个函数了,它将返回一个生成器对象。

2.1生成器的执行顺序

生成器的执行顺序:

  1. 在调用__next__()或者进入for循环之后,函数执行到yield就返回一个值,然后函数暂停。
  2. 在下次调用__next__()之后,生成器开始执行yield之后的语句。

例子:

def Gen(n):
    i=0
    while (i<n):
        yield i
        i=i+1
        yield 'step1'
        yield 'step2' 
#Gen(5)得到一个生成器对象        
c=Gen(5)
#执行结果
c.__next__()
>>0
c.__next__()
>>'step1'
c.__next__()
>>'step2'
c.__next__()
>>1
c.__next__()
>>'step1'
c.__next__()
>>'step2'
c.__next__()
>>2

2.2生成器对象的成员函数

在2.1的最后一个例子里面,如果查看生成器的帮助文档help(c),可以看到生成器对象有几个比较重要的成员函数:

  1. __next__()和__iter__():有这两个函数,生成器就具有了可迭代性
  2. send():这个函数很有用,可以用来给生成器函数传递一个新的值
  3. close():关闭这个生成器,关闭之后生成器就不可再调用
  4. throw():给生成器传入一个异常
2.2.1 send()

例子:

#send函数的用法
from random import randint

def Gen(n):
    i=0 
    while (i<n):
        i=i+1
        #yeild语句,在重新开始执行的时候可以获得一个值,这个值就是由send函数输入的
        res = yield randint(0,100)
        if res=='stop':
            print ('forced stop')
            break
if __name__=='__main__':
    #初始化一个生成器对象
    c=Gen(10)
    print (c.__next__())
    print (c.__next__())
    #传入‘start’字符串,
    print (c.send('start'))
    #传入‘stop’字符串,生成器跳出while循环
    print (c.send('stop'))
    #继续迭代生成器将报错
    print (c.__next__())

执行结果:
95
77
49
forced stop
Traceback (most recent call last):
  File "sample3.py", line 17, in <module>
    print (c.send('stop'))
StopIteration

从执行结果来看,

  • 调用前两个next和第三个send函数,生成器都会yield出结果。
  • 第四次调用send函数之后,也开始了迭代,不过由于退出了while循环,没有再一次回到yield,所以最终的结果里面只有三个数字
  • 最后一次调用next函数,出现报错,因为生成器已经关闭。
2.2.2 close()

例子:

#close函数的用法
from random import randint

def Gen(n):
    i=0 
    while (i<n):
        i=i+1
        yield randint(0,100)

if __name__=='__main__':
    c=Gen(10)
    print (c.__next__())
    print (c.__next__())
    #close生成器之后,再调用next将报错
    c.close()
    print (c.__next__())

执行结果:
81
65
Traceback (most recent call last):
  File "sample4.py", line 15, in <module>
    print (c.__next__())
StopIteration

close函数的用法就是关闭生成器对象,其实调用send或者next函数都会‘open’这个生成器,期间生成器都会记录执行的中间结果,直到生成器被关闭为止

2.2.3 throw()

例子:

#throw函数的用法
from random import randint

def Gen(n):
    i=0 
    while (i<n):
        i=i+1
        try:
            res = yield randint(0,100)
        except IOError:
            print ('get the IO Error')

if __name__=='__main__':
    c=Gen(10)
    print (c.__next__())
    #给生成器扔进去一个异常    
    c.throw(IOError)
    print (c.__next__())

执行结果:
39
get the IO Error
19

可以看到生成器内部成功的捕获并处理了这个异常,并且没有阻断生成器后面的执行流程

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 195,898评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,401评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,058评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,539评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,382评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,319评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,706评论 3 386
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,370评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,664评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,715评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,476评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,326评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,730评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,003评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,275评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,683评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,877评论 2 335

推荐阅读更多精彩内容