生成器yield和yield from

1、yield使用

1）函数中使用yield，可以使函数变成生成器。一个函数如果是生成一个数组，就必须把数据存储在内存中，如果使用生成器，则在调用的时候才生成数据，可以节省内存。

2）生成器方法调用时，不会立即执行。需要调用next()或者使用for循环来执行。使用for循环不需要自己捕获StopIteration异常。使用next()方法，当生产器方法执行结束会抛出StopIteration异常（只要不是使用yield返回数据，都会抛出StopIteration异常）。

示例：

def fib(max):
    n,a,b = 0,0,1
    while n < max:
        yield b
        a, b = b, a+b
        n = n + 1
    return 'done'


n = fib(10)
for n1 in n:
    print(n1)

3）yield不仅可以返回值，也可以接收值。下面面示例为生产消费模式。生产者生产一条记录，消费者消费一条记录。

4）调用生成器send方法传递数据时，必须先调用next(c)或者c.send(None)方法，执行到yield语句，等待接收数据。否则会报错。

以下代码为廖雪峰网站的示例

def consumer():
    r = ''
    while True:
        n = yield r
        if not n:
            return
        print('[CONSUMER] Consuming %s...' % n)
        r = '200 OK'




def produce(c):
    c.send(None)  # 和next方法一样 获取下一个值，必须先使用None参数调用一次， 执行到yield
    n = 0
    while n < 5:
        n = n + 1
        print('[PRODUCER] Producing %s...' % n)
        r = c.send(n)  # 先发送值给yield语句，再执行到yield语句时返回
        print('[PRODUCER] Consumer return:%s' % r)
    c.close()


c = consumer()
produce(c)

2、yield from的使用

1）为了让生成器（带yield函数），能简易的在其他函数中直接调用，就产生了yield from。

2）以下代码，htest为生成器，itest通过yield from 直接调用htest。这样itest也变成了一个生成器。创建itest实例不断的去获取数据，当生成器执行结束时，会抛出StopIteration异常。那这个异常是htest抛出的，还是itest抛出的。通过捕获异常，会发现其实是itest抛出异常，htest并不会抛出StopIteration异常。

3)yield from 也可以返回值，通过变量接收。变量接收的值，即htest使用return返回的值。示例代码中，当i==3时，会直接使用return返回，这时val的值就是100；因为htest函数中不是使用yield返回值，所以itest会继续执行print(val)语句。itest代码执行完，然而并没有使用yield返回数据（htest中没有，itest中也没有），所以马上会抛出StopIteration异常)(如果在itest函数最后使用yield返回一个数据，就不会抛出异常)。

def htest():
    i = 1
    while i < 4:
        n = yield i
        if i == 3:
            return 100
        i += 1


def itest():
    val = yield from htest()
    print(val)


t = itest()
t.send(None)
j = 0
while j < 3:
    j += 1
    try:
        t.send(j)
    except StopIteration as e:
        print('异常了')

简单的例子:

def fun_inner():
    i = 0
    while True:
        i = yield i


def fun_outer():
    yield from fun_inner()


if __name__ == '__main__':
    outer = fun_outer()
    outer.send(None)
    for i in range(5):
        print(outer.send(i))

参考链接:

https://www.cnblogs.com/cnkai/p/7514828.html

https://wenkefendou.gitbooks.io/python3-learning/content/yield_from.html

https://blog.csdn.net/chenbin520/article/details/78111399