一种神器一种优雅的迭代器——生成器

Posted 2021-01-19 mzfly

先看一段代码：

 

def fn(num):
    a = 0
    b = 1
    i = 0

    while i <= num:
        ret = a
        a, b = b, a + b
        i += 1
        yield ret


f1 = fn(15)

print(next(f1))
print(next(f1))
print(next(f1))
print(next(f1))
print(next(f1))

 

 

 

　　运行结果：

0
1
1
2
3

 

 

　　话说，在一个函数里有一个 while 循环，在循环里有一个 yield ，这个yield不同与云云丛生，它比闪电侠还牛批。它能让它所以在的函数瞬间变成一个生成器。

　　

一、生成器是一种特殊的迭代器，它比迭代器更优雅

　　什么是迭代器？看我上一篇文章：https://www.cnblogs.com/mzfly/p/9946908.html

 

二、生成器的分类　

　　1、函数式生成器
　　　　使用了 yield 的函数
　　　　next 运行到 yield，返回到yield后面的变量/值，并暂停执行下面的代码
　　　　下一次运行 next（）从next下一行开始执行

　　　　文章开头的 fn（） 函数就是一个函数式生成器

　　2、列表生成器
　　　　把列表生成式的 [ ] 改成 ( ) ——中括号改成小括号

　　　　比如：

　　　　　　 [i for i in range(10)] 是一个列表生产式，把它改成这样： (i for i in range(10)) ，那这就是一个生成器。

 

三、生成器的使用

　　有两种唤醒挂起的生成器的方式：

　　　　1、next（[生成器对象]）
　　　　2、生成器对象.send(数据)：
　　　　　　参数是要发送的数据
　　　　　　send()方法在使用过 next() 后再使用

 

 

四、利用生成器实现协程

def fn1():
  while True:
    print("xxx")
    yield
    有阻塞的代码

def fn2():
  while True:
    print("xxx")
    yield
    有阻塞的代码

f1 = fn1()
f2 = fn2()

next(f1)
...其它代码
next(f2)# 开始执行 f2 中的代码
...其它代码
next(f1)　　#又跳到 fn 内从 yield 后面的代码开始执行

...

　　　在代码执行的过程中，fn1 和 fn2 两个函数，是在同意个线执行的，但看上去并不是先后执行的关系，而是貌似是并行的。

　　在一个线程中的某个函数，可以在任何地方保存当前函数的一些临时变量等信息，然后切换到另外一个函数中执行，
注意不是通过调用函数的方式做到的，并且切换的次数以及什么时候再切换到原来的函数都由开发者自己确定。