python中yield的用法

原文鏈接：https://blog.csdn.net/mieleizhi0522/article/details/82142856

首先，如果你還沒有對yield有個初步分認識，那麼你先把yield看做“return”，這個是直觀的，它首先是個return，普通的return是什麼意思，就是在程序中返回某個值，返回之後程序就不再往下運行了。看做return之後再把它看做一個是生成器（generator）的一部分（帶yield的函數纔是真正的迭代器），好了，如果你對這些不明白的話，那先把yield看做return,然後直接看下面的程序，你就會明白yield的全部意思了：

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

就這麼簡單的幾行代碼就讓你明白什麼是yield，代碼的輸出這個：

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和return的關係和區別了，帶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))

再看一個這個生成器的send函數的例子，這個例子就把上面那個例子的最後一行換掉了，輸出結果：

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

先大致說一下send函數的概念：此時你應該注意到上面那個的紫色的字，還有上面那個res的值爲什麼是None，這個變成了7，到底爲什麼，這是因爲，send是發送一個參數給res的，因爲上面講到，return的時候，並沒有把4賦值給res，下次執行的時候只好繼續執行賦值操作，只好賦值爲None了，而如果用send的話，開始執行的時候，先接着上一次（return 4之後）執行，先把7賦值給了res,然後執行next的作用，遇見下一回的yield，return出結果後結束。

 

5.程序執行g.send(7)，程序會從yield關鍵字那一行繼續向下運行，send會把7這個值賦值給res變量

6.由於send方法中包含next()方法，所以程序會繼續向下運行執行print方法，然後再次進入while循環

7.程序執行再次遇到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組合成生成器進行實現，也可以用xrange(1000)這個生成器實現

yield組合：

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

輸出：

starting...
1
2
3
4
5
6
7
8
9
10

 xrange(1000):

for n in xrange(1000):
    a=n

 其中要注意的是python3時已經沒有xrange()了，在python3中，range()就是xrange()了，你可以在python3中查看range()的類型，它已經是個<class 'range'>了，而不是一個list了，畢竟這個是需要優化的。