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)使用更靈活