1. 程式人生 > 程式設計 >對python中return與yield的區別詳解

對python中return與yield的區別詳解

首先比較下return 與 yield的區別:

return:在程式函式中返回某個值,返回之後函式不在繼續執行,徹底結束。

yield: 帶有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

我直接解釋程式碼執行順序,相當於程式碼單步除錯:

.程式開始執行以後,因為foo函式中有yield關鍵字,所以foo函式並不會真的執行,而是先得到一個生成器g(相當於一個物件)

.直到呼叫next方法,foo函式正式開始執行,先執行foo函式中的print方法,然後進入while迴圈

.程式遇到yield關鍵字,然後把yield想想成return,return了一個4之後,程式停止,並沒有執行賦值給res操作,此時next(g)語句執行完成,所以輸出的前兩行(第一個是while上面的print的結果,第二個是return出的結果)是執行print(next(g))的結果,

.程式執行print("*"20),輸出20個

.又開始執行下面的print(next(g)),這個時候和上面那個差不多,不過不同的是,這個時候是從剛才那個next程式停止的地方開始執行的,也就是要執行res的賦值操作,這時候要注意,這個時候賦值操作的右邊是沒有值的(因為剛才那個是return出去了,並沒有給賦值操作的左邊傳引數),所以這個時候res賦值是None,所以接著下面的輸出就是res:None,

.程式會繼續在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出結果後結束。

.程式執行g.send(7),程式會從yield關鍵字那一行繼續向下執行,send會把7這個值賦值給res變數

.由於send方法中包含next()方法,所以程式會繼續向下執行執行print方法,然後再次進入while迴圈

.程式執行再次遇到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了,畢竟這個是需要優化的。

以上這篇對python中return與yield的區別詳解就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。