一、next()
带yield的函数是一个生成器,而不是一个函数了,这个生成器有一个函数就是next函数,next就相当于“下一步”生成哪个数,这一次的next开始的地方是接着上一次的next停止的地方执行的,下面例子帮助理解:
def test():
print("begin...")
while True:
a = yield 2
print("a:",a)
g = test()
print(next(g))
print("******************************")
print(next(g))
执行结果:
begin...
2
******************************
a: None
2
分析:
1)print(g=test()) 程序开始执行以后,因为test函数中有yield关键字,所以test函数并不会真正的执行,而是先得到一个生成器g;
2) print(next(g)) 直到调用next方法,test函数正式开始执行,先执行test函数中的print方法,然后进入while循环;
3)程序遇到yield关键字,然后把yield想成return,return了一个2之后,程序停止,并没有执行赋值给a操作,此时next(g)语句执行完成,所以输出前面两行(第一行是while上面的print的结果,第二行是retrun出来的结果);
4)print("******************************") 正常执行print;
5)print(next(g)) 又开始执行下面的print(next(g)),这个时候是从刚才next程序停止的地方开始执行,也就是要执行a的赋值操作,这个时候因为赋值操作的右边是没有值的,已经被return出去了这个时候a的值是none,所以接下来输出的是a: None
6)程序会继续在while里执行,又一次碰到yiled,这个时候同样return出2,然后程序停止。print函数输出的2就是这次return出的2
因此调用next的时候,生成器并不会从test函数的开始执行,只是接着上一步停止的地方开始,然后遇到yield后,return出要生成的数,此步就结束。
二、send()
send():从上次暂停的地方开始运行,并把值传进去(代码中的赋值),下面例子帮助理解:
def test():
print("begin...")
while True:
a = yield 8
print("a:",a)
g = test()
print(next(g))
print("******************************")
print(g.send(7))
执行结果:
begin...
8
******************************
a: 7
8
分析:
1)程序执行g.send(7),程序会从yield关键字那一行继续往下运行,send会把7这个值赋值给a变量;
2)由于send方法中包含next()方法,所以程序会继续往下运行执行Print方法,然后再次进入while循环;
3)程序再次遇到yield关键字,yield会返回后面的值后,程序再次暂停,直到再次调用next方法或send方法