python yield詳解

python yield詳解

如果你還沒有對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組合：（此段代碼裡面的 yiled相當于return）

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