天天看點

yieId詳解,以及和return的差別

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(相當于一個對象) 。

===》獲得技能:yield和函數差別?怎麼使用?(有yield的函數就不是函數了,是生成器,不會随意執行,想執行,要麼您調用next 要麼調用send方法,要麼您周遊)

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。

===》獲得技能:生成器下一次調用next/send方法時,從哪繼續執行?  (從剛才程式停止的下一步開始執行的,這裡下一步是該指派了) 

6.程式會繼續在while裡執行,又一次碰到yield,這個時候同樣return 出4,然後程式停止,print函數輸出的4就是這次return出的4。

小結:

到這裡你可能就明白yield和return的關系和差別了,帶yield的函數是一個生成器,而不是一個函數了,這個生成器有一個函數就是next函數,next就相當于“下一步”生成哪個數,

這一次的next開始的地方是接着上一次的next停止的地方執行的,是以調用next的時候,生成器并不會從foo函數的開始執行,隻是接着上一步停止的地方開始,然後遇到yield後,輸出要生成的數,此步就結束。

再來看個send的例子,此例與前例差別

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出結果後結束。

5.程式執行g.send(7),程式會從yield關鍵字那一行繼續向下運作,send會把7這個值指派給res變量。

6.由于send方法中包含next()方法,是以程式會繼續向下運作執行print方法,然後再次進入while循環。

7.程式執行再次遇到yield關鍵字,yield會傳回後面的值後,程式再次暫停,直到再次調用next方法或send方法。

yieId和return的異同

共同點:return和yield都用來傳回值;在一次性地傳回所有值場景中return和yield的作用是一樣的。

不同點:如果要傳回的資料是通過for等循環生成的疊代器類型資料(如清單、元組),return隻能在循環外部一次性地傳回,yeild則可以在循環内部逐個元素傳回。下邊我們舉例說明這個不同點。

為啥要用生成器:

#舉個例子
#用return
def squre(n):
    ls = [i*i for i in range(n)]
    return ls
for i in squre(5):
    print(i, end='  ')

      
#用yieId
def squre(n):
    for i in range(n):
        yield i*i
for i in squre(5):
    print(i, end='  ')      

輸出結果:

return例子:

0 1 4 9 16

yieId例子:

0 1 4 9 16

例子2

#比如說我要生成0到1000的數值

#你可能會這樣寫
for i in range(1000):
    print(i)

#預設生成一個含有1000個數的list了,是以很占記憶體

#我們可以用剛才的yield組合成生成器進行實作:(逐個生成,逐個輸出)


def foo(n):
    print('---start---')
    while n <= 1000:
        yield n
        n += 1
for n in foo(0):
    print(n)      

結論:

yield 生成器相比 return一次傳回所有結果的優勢:

(1)反應更迅速

(2)更節省空間

(3)使用更靈活