函数生成器

生成器

• 生成器 generator
  • 生成器指的是生成器对象，可以由生成器表达式得到，也可以使用yield关键字得到一个生成器函数，调用这个函数得到一个生成器对象
  • 生成器对象，是一个可迭代对象，是一个迭代器
  • 生成器对象，是延迟计算、惰性求值的
• 生成器函数
  • 函数体中包含 yield语句的函数，就是生成器函数，调用后返回生成器对象

---

• 普通函数调用，函数会立即执行直到执行完毕,生成器函数调用，并不会立即执行函数体，而是需要使用next函数来驱动生成器函数执行后获得的生成器对象。
• 生成器表达式和生成器函数都可以得到生成器对象，只不过生成器函数可以写的更加复杂的逻辑

生成器的执行

• 在生成器函数中，可以多次yield，每执行一次yield后会暂停执行，把yield表达式的值返回
• 再次执行会执行到下一个yield语句又会暂停执行
• return语句依然可以终止函数运行，但return语句的返回值不能被获取到
• return会导致当前函数返回，无法继续执行，也无法继续获取下一个值，抛出StopIteration异常如果函数没有显式的return语句，如果生成器函数执行到结尾（相当于执行了return None），一样会抛出StopIteration异常

生成器函数

• 包含yield语句的生成器函数调用后，生成生成器对象的时候，生成器函数的函数体不会立即执行
• next(generator) 会从函数的当前位置向后执行到之后碰到的第一个yield语句，会弹出值，并暂停函数执
• 再次调用next函数，和上一条一样的处理过程
• 继续调用next函数，生成器函数如果结束执行了（显式或隐式调用了return语句），会抛出StopIteration异常

生成器应用

1.无限循环

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

2、计数器

def inc():
    def counter():
        i = 0
        while True:
            i += 1
            yield i    
    c = counter()
    def inner():
        return next(c)    
    return inner # return lambda : next(c)
foo = inc()
print(foo()) # 打印什么?
print(foo()) # 打印什么?
print(foo()

# 代码中的inner函数可以由lambda表达式替代

3.斐波那契数列

def fib():    
    x = 0    
    y = 1    
    while True:
        yield y
        x, y = y,x + y
foo = fib()
for i in range(10):    # 打印前10行
    print(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())
print(foo())

打印结果 :
1
2
3
1
2
3
4

• 调用send方法，就可以把send的实参传给yield语句做结果，这个结果可以在等式右边被赋值给其它变量
• send和next一样可以推动生成器启动并执行

5.协程 Coroutine

• 生成器的高级用法
• 它比进程、线程轻量级，是在用户空间调度函数的一种实现
• Python3 asyncio就是协程实现，已经加入到标准库
• Python3.5 使用async、await关键字直接原生支持
• 协程协程调度器实现思路有
  • 2个生成器A、B
  • next(A)后，A执行到了yield语句暂停，然后去执行next(B)，B执行到yield语句也暂停，然后再次调用
  • next(A)，再调用next(B)在，周而复始，就实现了调度的效果
  • 可以引入调度的策略来实现切换的方式
• 协程是一种非抢占式调度

---

• yield from语法
  • yield from iterable等价于for item in iterable: yield item。

def inc():    
    for x in range(1000):
        yield x
# 使用yield from 简化
def inc():    
    yield from range(1000)
foo = inc()
print(next(foo))
print(next(foo))
print(next(foo))

本质上yield from的意思就是，从from后面的可迭代对象中拿元素一个个yield出去。