生成器

Posted 2021-02-09 selling-fish-bears

两种创建生成器的方法：

　　将列表推导式的 [ ] 换成 ( )

　　在函数里使用 yield

三种启动生成器的方法：

　　next()

　　send(参数)

　　for循环

迭代器和生成器的区别：

　　迭代器可以减少资源耗用，实现循环

　　生成器是特殊的迭代器，可以通过yield暂停执行，也可以通过send启动方式传参改变执行方式

创建生成器对象和实例化一样，互不影响

 

实例

def fibo(all_num):
    """存在yield的函数，会变为生成器的模板"""

    a, b = 0, 1
    current_num = 0
    while current_num < all_num:
        ret = yield a  # yield暂停，返回a的值，并且下次遍历时从yield下一个语句开始，即赋值给ret
        print("ret：", ret)
        a, b = b, a+b
        current_num += 1


# 创建生成器对象
obj = fibo(10)
# obj2是新开辟的内存空间，与ojb互不影响
obj2 = fibo(10)

# 启动生成器方法1
ret = next(obj)
print("obj:", ret)

# 启动生成器方法2，可传参，但第一次启动就使用send，参数必须是None
ret = obj.send("熊")  # send(参数)中的参数等价于生成器里的yidel a的结果
print("obj:", ret)

# 启动生成器方法3
for i in obj:
    print(i)