python中yield的用法簡明教程

python中yield關鍵字是區分新老pythoner的標誌之一。

yield是乙個六級詞彙，常見意思有「產量，屈服(動詞)」。這裡的yield大概率解釋為乙個僻義「繳出」。咱們在什麼情況下，會用到yield呢？答：處理大型資料集的時候。

yield單獨解釋比較費勁，但是跟next()結合看，就會非常簡單。先看乙個小栗子:

a = [1, 2, 3]
def y1():
for i in a:
yield i
b = y1()
print(b)
print(next(b))
print(next(b))
print(next(b))

輸出為：

1

23

咱們用乙個yield構建出乙個生成器函式y1，a是咱們要遍歷的物件。咱們直接print生成器的返回值，得到的是乙個記憶體位址。咱們發現，next(b)每次執行一下，yield指向的記憶體位址就會順序後移。而next(b)的返回值是當前位址內的資料。yield+next就是乙個順序迭代子。

常見如pytorch，keras等nn框架處理資料集的基礎類都會用到yield+next來獲取批資料。這樣有什麼優勢呢，首先可以把資料入口封裝成介面的形式，模組性更好，也不會讓資料在main函式中「喧賓奪主」。其次，預處理資料等工作都可以放到生成器的部分。

python中yield的用法 簡明教程