首先,拿好寶劍: 先把yield看做“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
按順序逐句解釋,你絕對就清楚明白,獨步天下了:
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方法。
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
那么問題來了,為什么要用生成器?
生成器相比一次列出所有的內容的有優勢:
- 更節省存儲空間
- 響應更迅速
- 使用更靈活
不廢話,翠花上栗子,直觀感受:
比如說我要100個數字 0,1,2,3,4,5,6............1000
你可能會這樣寫:
for i in range(1000):
print(i)
range(1000)就默認生成一個含有1000個數的list了,所以很占內存。
我們可以用剛才的yield組合成生成器進行實現:(逐個生成,逐個輸出)
def foo(n):
print('---start---')
while n <= 1000:
yield n
n += 1
for n in foo(0):
print(n)
好了,就到這了,希望諸位劍術有所變強,江湖再見。
