快捷導(dǎo)航

一篇文章快速理解python中的yield關(guān)鍵字

更新時間：2024年11月21日 09:27:11 作者：moz與京

Python中的yield關(guān)鍵字可以讓函數(shù)變成生成器,產(chǎn)生一個值后暫停,下次調(diào)用時從上次停下的地方繼續(xù)執(zhí)行,從而節(jié)省內(nèi)存并提高效率,這篇文章主要介紹了python中yield關(guān)鍵字的相關(guān)資料,需要的朋友可以參考下

前言

想象一下，你正在做一個蛋糕，但是這個蛋糕非常大，你不想一次性把它全部做完，而是想做一點吃一點，yield 關(guān)鍵字就能讓你做到這一點。

（插一句，yield本身最廣泛使用的意思就是：產(chǎn)生、產(chǎn)量，其實就可以反映它的作用）

一、yield 的基本作用

在 Python 中，yield 關(guān)鍵字可以讓我們的函數(shù)變成一個“生成器”。生成器就像一個特殊的函數(shù)，它可以記住自己做到哪一步了，然后下次從上次停下的地方繼續(xù)做。

二、如何使用 yield

把函數(shù)變成生成器：只要你在函數(shù)里用到了 yield，這個函數(shù)就不再是一個普通的函數(shù)了，它變成了一個生成器。
產(chǎn)生值：每次函數(shù)執(zhí)行到 yield 時，它會“產(chǎn)出”一個值，然后停下來。下次再調(diào)用這個生成器的時候，它會從上次停下的地方繼續(xù)執(zhí)行。
記住狀態(tài)：當(dāng)生成器停下來的時候，它會記住所有的變量和它們的狀態(tài)，這樣下次開始的時候，它就能從上次停下的地方繼續(xù)。

三、一個簡單的例子

def count_to(max):
    count = 1
    while count <= max:
        yield count  # 還沒結(jié)束但是先記著，這里用 yield 產(chǎn)生值
        count += 1   # 繼續(xù)

# 創(chuàng)建生成器
counter = count_to(5)

# 迭代生成器
for number in counter:
    print(number)

這段代碼會輸出：