带有yield的函数,就是一个生成器,yield与return类似,都会返回一个参数,但是最大的不同点是return值后就不会再运行程序,但是yield会在下次调用生成器是接着执行
def foo():
print("starting...")
while True:
res = yield 4
print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(next(g))
输出:
starting...
4
********************
res: None
4
代码的执行顺序:
1、因foo()函数中有yield关键字,所以foo()函数并不会真的执行,而是先得到一个生成器g对象
2、直到调用next()方法后,foo()函数开始执行,先执行foo()函数中print方法,进入while循环
3、程序遇到yield关键字,然后把yield想象成return,return一个4后,程序暂时停止,并未给res赋值,此时next(g)语句执行完成,所以输出的前两行(第一个是while上面的print的结果,第二个是return的结果)是执行print(next(g))的结果
4、程序执行print("*"*20),输出20个*
5、执行下面的print(next(g)),这个时候和上面那个差不多,不过不同的是,这个时候是从刚才那个next程序停止的地方开始执行的,也就是要执行res的赋值操作,这时候要注意,这个时候赋值操作的右边是没有值的(因为刚才那个是return出去了,并没有给赋值操作的左边传参数),所以这个时候res赋值是None,所以接着下面的输出就是res:None
6、继续在while里执行,又一次碰到yield,这个时候同样return 出4,然后程序停止,print函数输出的4就是这次return出的4
yield生成器的特点:
- yield 是一个类似 return的关键字,迭代一次遇到yield时就返回yield后面的值。重点是:下一次迭代时,从上一次迭代遇到的yield后面的代码开始执行
- 带有 yield 的函数不再是一个普通函数,而是一个生成器generator,可用于迭代
- 生成器是可以迭代的,但只可以读取它一次。因为用的时候才生成
- yield就是 return 返回一个值,并且记住这个返回的位置,下次迭代就从这个位置后开始
- 一个生成器只能只能用一次:将一个生成器中的值全部取出来之后,再从中取值得到的是空
- 在生成器表达式外有for循环时,把for循环拆开来分析
- send(msg)与next()的区别在于send可以传递参数给yield表达式,这时传递的参数会作为yield表达式的值,而yield的参数是返回给调用者的值。——换句话说,就是send可以强行修改上一个yield表达式值。比如函数中有一个yield赋值,a = yield 5,第一次迭代到这里会返回5,a还没有赋值。第二次迭代时,使用.send(10),那么,就是强行修改yield 5表达式的值为10,本来是5的,那么a=10
def demo():
for i in range(4):
yield i
g=demo()
g1=(i for i in g)
g2=(i for i in g1)
print(list(g1))
print(list(g2))
1、在执行list(g1)之前,生成器函数,生成器表达式中的内容都没有执行。(只有从生成器中取值时,生成器函数,生成器表达式中的内容才会执行)
2、当执行list(g1)时,就是从g1这个生成器中取值(数据类型强转),g1又去找g要值,所以执行结果是:[0,1,2,3]
3、当执行list(g2)时,g2生成器会找g1生成器要值,而list(g1)语句,以及将生成器g1中的值取出来,所以g1是空,执行结果为:[]
def add(n,i):
return n+i
def test():
for i in range(4):
yield i
g=test()
for n in [1,10]:
g=(add(n,i) for i in g)
print(list(g))
同上:在list(g)之前,生成器函数,生成器表达式内的语句都没有执行。只有当取生成器中的值得时候才执行。
遇见for循环里面有生成器表达式,将for循环拆解开来
n = 1
g = (add(n,i) for i in g)
n = 10
g = (add(n,i) for i in g)
当执行list(g)时,生成器函数或者是生成器表达式中的内容开始执行。
n就变成了循环中的最后一个值。
for循环中就变为:
n = 10
g = (add(n,i) for i in (add(n,i) for i in test()))
结果为:[20,21,22,23]