生成器

生成器指的是生成器对象,可以有生成器表达式获得,也可以由yield关键字得到一个生成器,调用这个函数得到一个生成器对象。生成器对象是一个可迭代对象,是一个迭代器。生成器对象是延迟计算、惰性求值的。

生成器函数

函数体中包含yield语句的函数,就是生成器函数,调用该函数返回一个生成器对象。

m = (i for i in range(5)) # 生成器表达式
next(m)
next(m)

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

# 生成器函数:
def inc():
    for i in range(5):
        yield i
        
print(type(inc)) # <class 'function'>
print(type(inc())) # <class 'generator'>

g = inc()
print(type(g)) # <class 'generator'>
print(next(g)) # 0
for x in g:
    print(x) # 1 2 3 4
for y in g:
    print(y) 

普通函数调用,函数会立即执行到执行完毕。生成器函数调用,并不会立即执行,而是需要使用next函数来驱动其执行生成器函数调用后获得的生成器对象。

生成器函数相对于生成器表达式的优势在于:生成器的代码可以写的更加复杂。

生成器的执行

def gen():
    print('line 1')
    yield 1
    print('line 2')
    yield 2
    print('line 3')
    return 3
    yield 4

next(gen()) # line 1
next(gen()) # line 1
g = gen() # line 1
print(next(g)) # 1 & line 2 
print(next(g)) # 2 & line 3
# print(next(g)) # StopIteration: 3
print(next(g, 'end')) # 生成器没有元素了,给个默认值
  • 在生成器函数中,可以多次yield,每执行一次yield后会暂停执行,把yield表达式的值返回
  • 再次执行会执行到下一个yield又会暂停
  • return语句依然可以终止函数运行,但return的值不可能被捕获到
  • return会导致当前函数返回,无法继续执行也无法继续获取下一个值,抛出StopIteration异常
  • 如果函数没有显示的return语句,如果生成器执行到了结尾(相当于return None),一样会抛出StopIteration异常

总结:

  1. 包含yield语句的生成器函数调用后,生成生成器对象的时候,函数体并不会执行(也就是,g = gen()不会使得 print('line 1') 执行)
  2. next(generator)会从函数的当前位置向后执行,直到碰到yield语句,会弹出值,并暂停函数运行
  3. 再次调用next(),和上面一样
  4. 继续调用next,如果生成器结束执行了(显示/隐式调用return语句),会抛出StopIteration异常

生成器应用

1、无限循环

def counter():
    i = 0
    while True:
        i += 1
        yield i
        
c = counter()
next(c)
next(c)
next(c)
...

2、计数器

def inc():
    def counter():
        i = 0
        while True:
            i += 1
            yield i

    c = counter()
    return next(c)

print(inc()) # 1
print(inc()) # 1
print(inc()) # 1

思考:为什么都是返回1?-- 因为函数的调用都是独立的,每次的inc()都使用一个新的counter ,所以每次都能返回1。

修改:

def inc():
    def counter():
        i = 0
        while True:
            i += 1
            yield i
            
    c= counter()
    def inner():
        return next(c)
    return inner # 可以用lambda函数代替

foo = inc()
print(foo()) # 1
print(foo()) # 2 
print(foo()) # 3

思考:为什么能够打印1,2,3? -- 因为inc()这次返回的是一个函数对象而非常量,且这个函数对象是在inc内部定义的,所以inc函数不会在栈内释放,即counter函数,c,inner函数都在inc函数的栈帧中;调用foo,就是调用inner函数,inner函数返回 next(c),c 是 inner函数外部的自由变量 (闭包),c 是唯一的且被next()多次调用,所以能返回1 2 3。

3、斐波那契数列

def fib():
    x = 0
    y = 1
    while True:
        yield y
        x, y = y, x+y
        
foo = fib()
for i in range(100):
    next(foo)

4、生成器交互

python提供了一个和生成器对象交互的方法send,该方法可以和生成器沟通。

# 带重置功能的计数器
def inc():
    def counter():
        i = 0
        while True:
            i += 1
            response = yield i 
            if response is not None: # 重置功能
                i = response
            
    c= counter()
    return lambda x=False: next(c) if not x else c.send(0) 
    
foo = inc()
print(foo())
print(foo())
print(foo())
print(foo(True))
print(foo())
print(foo())
  • 调用生成器的send方法,可以将send的实参传给yield做结果,这个结果可以在等式右边被赋值给其他变量
  • send 和 next 一样,可以推动生成器执行

5、协程Coroutine

  • 生成器的高级语法
  • 它比线程、进程轻量级,实在用户空间的调度函数的一种体现
  • Python3 asyncio 就是协程实现,已经加入到标准库
  • python 3.5 使用 async、await 关键字支持原生协程
  • 协程调度器实现思路:
    • 有两个生成器A、B
      • next(A) 之后,A执行到了yield语句暂停,然后去执行next(B),B执行到yield语句也暂停,然后再次调用next(A),再调用next(B),周而复始,就实现了调度的效果
      • 可以引入调度的策略来实现切换的方式
  • 协程是一种非抢占式调度,只要没有遇到yield,就能一直执行下去,不会被抢占运行的权力

yield from 语法

从python3.3开始增加了yield语法,使得 yield from iterable 等价于 for item in iterable: yield item

yield from 就是一种语法糖。本质上就是在 iterable 中拿元素一个个 yield 出去。

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

推荐阅读更多精彩内容