多任务--协程--生成器 yield send

创建生成器的方法1：把列表生成的 [] 换成 （）。用的比较少。

创建生成器的方法2：定义一个函数，让这个函数变成生成器。只要函数中有 yield 就是生成器。

使用生成器完成 feibonacii 数列

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj = creat_num(10)

# 生成器是可以迭代的
for num in obj:
    print(num)

因为生成器是特殊的迭代器，所以可以使用 next（）
生成器中的 yield a 相当于返回值为a ，当创建生成器对象并且调用时，运行到yield语句暂停，并且返回yield的值，当再次调用时，开始从yield语句往下循环直到yield语句再次暂停，并且将新的值返回。下面使用 next（）语句来看一下生成器的运行。

def creat_num(all_num):
    print('---1---')
    a, b = 0, 1
    current_num = 0
    print('---2---')
    while current_num < all_num:
        print('---3---')
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1
        print('---4---')


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj = creat_num(10)

ret = next(obj)
print(ret)

ret = next(obj)
print(ret)

如果创建两个生成器对象，调用时会不会相互影响：

def creat_num(all_num):
    print('---1---')
    a, b = 0, 1
    current_num = 0
    print('---2---')
    while current_num < all_num:
        print('---3---')
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1
        print('---4---')


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj1 = creat_num(10)
obj2 = creat_num(10)

ret = next(obj1)
print(ret)

ret = next(obj1)
print(ret)

ret = next(obj2)
print(ret)

ret = next(obj1)
print(ret)

ret = next(obj1)
print(ret)

可以看出，创建的两个生成器对象之间互不影响。
但是在运行中会发现一个问题，当生成器已经结束，再去取值时，会产生异常 StopIteration。

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj1 = creat_num(2)

while True:
    ret = next(obj1)
    print(ret)

加入一个异常判断：

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj1 = creat_num(2)

while True:
    try:
        ret = next(obj1)
        print(ret)
    except Exception as ret:
        break

那如果生成器中有return 语句，那么在调用生成器对象时，怎么样？

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1
    return 'ok....'


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj1 = creat_num(2)

while True:
    try:
        ret = next(obj1)
        print(ret)
    except Exception as ret:
        break

可以看出捕获不到return的返回值。那么怎么才能捕获呢：当异常时，调用ret.value

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a # 如果一个函数中有yield语句，那么这个就不是函数，而是一个生成器的模板
        a, b = b, b + a
        current_num += 1
    return 'ok....'


# 如果在调用creat_num的时候，发现这个函数中有yield，那么此时不是调用函数，而是创建一个生成器对象
obj1 = creat_num(2)

while True:
    try:
        ret = next(obj1)
        print(ret)
    except Exception as ret:
        print(ret.value)
        break

我们除了使用next 来唤醒生成器继续执行，还可以使用send。使用send的一个好处是：可以在唤醒的同时向断点处传入一个附加数据。

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a
        a, b = b, b + a
        current_num += 1


obj1 = creat_num(10)

ret = next(obj1)
print(ret)

ret = next(obj1)
print(ret)

使用send来唤醒生成器：

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        yield a
        a, b = b, b + a
        current_num += 1


obj1 = creat_num(10)

ret = next(obj1)
print(ret)

ret = obj1.send(None)
print(ret)

效果是一样的。唯一的区别就是send 能传入值。当不传入时，必须写入None ，啥也不写会报错，要注意。

使用send传入一个值，相当于把这个值赋给yield a ，所以必须将yield a复制给一个参数。

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        ret = yield a
        print('--ret---',ret)
        a, b = b, b + a
        current_num += 1


obj1 = creat_num(10)

ret = next(obj1)
print(ret)

ret = obj1.send('hahahaha')
print(ret)

这里必须注意，一般不能将send 赋值 位于距 用在取生成器的第一个值。因为程序从头开始运行，不能讲send 赋值 传送给yield 。

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        ret = yield a
        print('--ret---',ret)
        a, b = b, b + a
        current_num += 1


obj1 = creat_num(10)

ret = obj1.send('hahahaha')
print(ret)

ret = next(obj1)
print(ret)

ret = obj1.send('hahahaha')
print(ret)

那我就是非要用send 启动，怎么办？ 传入None 。一般来说，无论是启动还是唤醒，都用next 。除非有参数要传入时，才要使用send。

def creat_num(all_num):
    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        ret = yield a
        print('--ret---',ret)
        a, b = b, b + a
        current_num += 1


obj1 = creat_num(10)

ret = obj1.send(None)
print(ret)

ret = next(obj1)
print(ret)

ret = obj1.send('hahahaha')
print(ret)

可迭代：能使用for 循环
迭代器：有一个iter方法 和一个 next方法的对象
生成器：是一种特殊的迭代器，可以没有iter方法和next方法。只要保证有用yield就可以。
迭代器和生成器保存的不是数据，而是生成数据的方法，所以占空间小。
生成器的特点是：可以程序执行一部分保存，然后返回。在调用时，接着执行。

迭代器：减少内存空间，能实现循环。
生成器：能够让一个函数暂停执行并且返回。