1. 程式人生 > 程式設計 >Python生成器next方法和send方法區別詳解

Python生成器next方法和send方法區別詳解

yield的語法規則是:

在yield這裡暫停函式執行,並返回yield後面表示式的值(預設為None),直到被next()再次呼叫時,從上次暫停的yield程式碼處繼續往下執行。當沒有可繼續next()時,丟擲異常,該異常可被for迴圈處理。

def fib(n):

def fib(n):
  a,b = 0,1
  i = 0
  while i < n:
    yield b
    a,b = b,a+b
    i += 1

if __name__ == '__main__':
  f = fib(10)
  for item in f:
    print(item)

python中,含有yield關鍵字的物件就是一個生成器,每次呼叫next方法時會執行到yield後面的語句,然後返回yield後面程式碼塊的執行結果

def foo():
  bar_a = yield 1     # bar_a是語句塊(yield 1)的返回值,預設為None
  bar_b = yield bar_a
  yield "最後一個值,再迭代就要報StopIteration了"

f = foo()          # 建立生成器,此時沒有執行foo()裡的任何語句
print(next(f))       # 從foo()裡進入,一直執行到(yield 1)處,此時變數bar_a還沒有建立
print(next(f))       # 先將語句塊(yield 1)的返回值賦值個bar_a,此時bar_a的值是None。
              # 然後執行到語句塊(yield bar_a),bar_b也還沒有被建立
print(next(f)

>>>1
>>>None
>>>最後一個值,再迭代就要報StopIteration了

send()

def foo():
  bar_a = yield 1
  bar_b = yield bar_a
  yield "最後一個值,再迭代就要報StopIteration了"

f = foo()
print(f.send(None))    # 和next(f)的效果一模一樣
print(f.send("my lover")) # 覆蓋yield 1語句的返回值
print(next(f))

>>>1
>>>my lover
>>>最後一個值,再迭代就要報StopIteration了

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。