Python从门到精通(三):高级操作-03-yield关键字

可以简单理解来return。但它还有其它的用途。详细如下：

一、当做return使用

它首先是个return，就是在程序中返回某个值，返回之后程序就不再往下运行了。看做return之后再把它看做一个是生成器（generator）的一部分（带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

直接解释代码运行顺序，相当于代码单步调试：

程序开始执行以后，因为foo函数中有yield关键字，所以foo函数并不会真的执行，而是先得到一个生成器g；

直到调用next方法，foo函数正式开始执行，先执行foo函数中的print方法，然后进入while循环；

程序遇到yield关键字，然后把yield想像成return，return了一个4之后，程序停止，并没有执行赋值给res操作，此时next(g)语句执行完成；输出starting和4；

程序执行print("*"*20)，输出20个*

又开始执行下面的print(next(g))，这个时候和上面那个差不多，不过不同的是，这个时候是从刚才那个next程序停止的地方开始执行的，也就是要执行res的赋值操作，这时候要注意，这个时候赋值操作的右边是没有值的（因为刚才那个是return出去了，并没有给赋值操作的左边传参数），所以这个时候res赋值是None，所以输出就是res:None；

程序会继续在while里执行，又一次碰到yield，这个时候同样return 出4，然后程序停止，print函数输出的4就是这次return出的4；

二、生成器应用

带yield的函数是一个生成器，而不是一个函数，这个生成器有一个函数就是next函数，next就相当于“下一步”生成哪个数，这一次的next开始的地方是接着上一次的next停止的地方执行的，所以调用next的时候，生成器并不会从foo函数的开始执行，只是接着上一步停止的地方开始，然后遇到yield后，return出要生成的数，此步就结束：

def foo():
print("starting...")
while True:
res = yield 4
print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(g.send(7))

starting...
4
********************
res: 7
4

先大致说一下send函数的概念：还有上面那个res的值为什么是None，这个变成了7，这是因为，send是发送一个参数给res的，因为上面讲到，return的时候，并没有把4赋值给res，下次执行的时候只好继续执行赋值操作，只好赋值为None了，而如果用send的话，开始执行的时候，先接着上一次（return 4之后）执行，先把7赋值给了res,然后执行next的作用，遇见下一回的yield，return输出结果后结束。

程序执行g.send(7)，程序会从yield关键字那一行继续向下运行，send会把7这个值赋值给res变量；

由于send方法中包含next()方法，所以程序会继续向下运行执行print方法，然后再次进入while循环；

程序执行再次遇到yield关键字，yield会返回后面的值后，程序再次暂停，直到再次调用next方法或send方法；

三、节省内存

为什么用这个生成器，是因为如果用List的话，会占用更大的空间，比如说取0,1,2,3,4,5,6............1000，你可能会这样：

for n in range(1000):
a=n

这个时候range(1000)就默认生成一个含有1000个数的list了，所以很占内存。这个时候你可以用刚才的yield组合成生成器进行实现：

def foo(num):
print("starting...")
while num<10:
num=num+1
yield num
for n in foo(0):
print(n)