Python中Yield的基本用法及Yield與return的區別解析

2022-11-01 14:02:40

一、yield的功能

Python中有一個非常有用的語法叫做生成器，用到的關鍵字就是yield。yield的作用是把一個函數變成生成器(generator)，起到一個延遲的作用。
帶有yield的函數在Python中被稱之為generator(生成器)，也就是說，當你呼叫這個函數的時候，函數內部的程式碼並不立即執行，這個函數只是返回一個生成器(Generator Iterator)。

1. 使用next方法迭代生成器

generator函數怎麼呼叫呢？答案是next函數。

def generator():
 for i in range(10) :
  yield i*i
  
gen = generator()
print(gen)
  
print("first iteration:")
print(next(gen))
  
print("second iteration:")
print(next(gen))
  
print("third iteration:")
print(next(gen))
  
print("fourth iteration:")
print(next(gen))

程式輸出:

<generator object generator at 0x7ffaad115aa0>first iteration:
0
second iteration:
1
three iteration:
4
four iteration:
9

在函數第一次呼叫next(gen)函數時，generator函數從開始執行到yield，並返回yield之後的值。

在函數第二次呼叫next(gen)函數時，generator函數從上一次yield結束的地方繼續執行，直至下一次執行到yield的地方，並返回yield之後的值。依次類推。

2. 使用send()方法與生成器函數通訊

def generator():
 x = 1
 while True:
  y = (yield x)
  x += y
  
gen = generator()
    
print("first iteration:")
print(next(gen))
  
print("send iteration:")
print(gen.send(10))

程式碼輸出：

first iteration:
1
send iteration:
11

生成器(generator)函數用yield表示式將處理好的x傳送給生成器(Generator)的呼叫者；然後生成器(generator)的呼叫者可以通過send函數，將外部資訊替換生成器內部yield表示式的返回值，並賦值給y，並參與後續的迭代流程。

範例2：

def dog(name):
    food_list=[]
    print('狗哥%s準備開吃'%name)
    while True:
        food=yield food_list   #  暫停food=yield='一桶泔水'
        print('狗哥%s吃了%s'%(name,food))
        food_list.append(food)
 
dog_name=dog('Ami')
next(dog_name)
dog_name.send('肉肉')
dog_name.send('包子')

程式碼輸出：

狗哥Ami準備開吃
狗哥Ami吃了肉肉
狗哥Ami吃了包子

3. Yield的好處

Python之所以要提供這樣的解決方案，主要是記憶體佔用和效能的考量。看類似下面的程式碼：

def my_range(start,stop,step=1):
    while start<stop:
        yield start #暫停
        start += step
 
g=my_range(1,30,2)  #1 3
print('first iteration:')
print(next(g))
print('scond iteration:')
print(next(g))
print(next(g))

執行結果