快捷導(dǎo)航

python中yield函數(shù)的用法詳解

更新時(shí)間：2022年02月10日 09:47:28 作者：馮爽朗

這篇文章主要為大家詳細(xì)介紹了python中yield函數(shù)的用法，數(shù)據(jù)庫(kù)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

首先，如果你還沒(méi)有對(duì)yield有個(gè)初步分認(rèn)識(shí)，那么你先把yield看做“return”，這個(gè)是直觀的，它首先是個(gè)return，普通的return是什么意思，就是在程序中返回某個(gè)值，返回之后程序就不再往下運(yùn)行了?？醋鰎eturn之后再把它看做一個(gè)是生成器（generator）的一部分（帶yield的函數(shù)才是真正的迭代器），好了，如果你對(duì)這些不明白的話，那先把yield看做return,然后直接看下面的程序，你就會(huì)明白yield的全部意思了：

def foo():
    print("starting...")
    while True:
        res = yield 4
        print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(next(g))

就這么簡(jiǎn)單的幾行代碼就讓你明白什么是yield，代碼的輸出這個(gè)：

starting...
4
********************
res: None
4

我直接解釋代碼運(yùn)行順序，相當(dāng)于代碼單步調(diào)試：

1.程序開(kāi)始執(zhí)行以后，因?yàn)閒oo函數(shù)中有yield關(guān)鍵字，所以foo函數(shù)并不會(huì)真的執(zhí)行，而是先得到一個(gè)生成器g(相當(dāng)于一個(gè)對(duì)象)

2.直到調(diào)用next方法，foo函數(shù)正式開(kāi)始執(zhí)行，先執(zhí)行foo函數(shù)中的print方法，然后進(jìn)入while循環(huán)

3.程序遇到y(tǒng)ield關(guān)鍵字，然后把yield想想成return,return了一個(gè)4之后，程序停止，并沒(méi)有執(zhí)行賦值給res操作，此時(shí)next(g)語(yǔ)句執(zhí)行完成，所以輸出的前兩行（第一個(gè)是while上面的print的結(jié)果,第二個(gè)是return出的結(jié)果）是執(zhí)行print(next(g))的結(jié)果，

4.程序執(zhí)行print("*"*20)，輸出20個(gè)*

5.又開(kāi)始執(zhí)行下面的print(next(g)),這個(gè)時(shí)候和上面那個(gè)差不多，不過(guò)不同的是，這個(gè)時(shí)候是從剛才那個(gè)next程序停止的地方開(kāi)始執(zhí)行的，也就是要執(zhí)行res的賦值操作，這時(shí)候要注意，這個(gè)時(shí)候賦值操作的右邊是沒(méi)有值的（因?yàn)閯偛拍莻€(gè)是return出去了，并沒(méi)有給賦值操作的左邊傳參數(shù)），所以這個(gè)時(shí)候res賦值是None,所以接著下面的輸出就是res:None,

6.程序會(huì)繼續(xù)在while里執(zhí)行，又一次碰到y(tǒng)ield,這個(gè)時(shí)候同樣return 出4，然后程序停止，print函數(shù)輸出的4就是這次return出的4.

到這里你可能就明白yield和return的關(guān)系和區(qū)別了，帶yield的函數(shù)是一個(gè)生成器，而不是一個(gè)函數(shù)了，這個(gè)生成器有一個(gè)函數(shù)就是next函數(shù)，next就相當(dāng)于“下一步”生成哪個(gè)數(shù)，這一次的next開(kāi)始的地方是接著上一次的next停止的地方執(zhí)行的，所以調(diào)用next的時(shí)候，生成器并不會(huì)從foo函數(shù)的開(kāi)始執(zhí)行，只是接著上一步停止的地方開(kāi)始，然后遇到y(tǒng)ield后，return出要生成的數(shù)，此步就結(jié)束。

****************************************************************************************************************************************

def foo():
    print("starting...")
    while True:
        res = yield 4
        print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(g.send(7))

再看一個(gè)這個(gè)生成器的send函數(shù)的例子，這個(gè)例子就把上面那個(gè)例子的最后一行換掉了，輸出結(jié)果：

starting...
4
********************
res: 7
4

先大致說(shuō)一下send函數(shù)的概念：此時(shí)你應(yīng)該注意到上面那個(gè)的紫色的字，還有上面那個(gè)res的值為什么是None，這個(gè)變成了7，到底為什么，這是因?yàn)?，send是發(fā)送一個(gè)參數(shù)給res的，因?yàn)樯厦嬷v到，return的時(shí)候，并沒(méi)有把4賦值給res，下次執(zhí)行的時(shí)候只好繼續(xù)執(zhí)行賦值操作，只好賦值為None了，而如果用send的話，開(kāi)始執(zhí)行的時(shí)候，先接著上一次（return 4之后）執(zhí)行，先把7賦值給了res,然后執(zhí)行next的作用，遇見(jiàn)下一回的yield，return出結(jié)果后結(jié)束。

5.程序執(zhí)行g(shù).send(7)，程序會(huì)從yield關(guān)鍵字那一行繼續(xù)向下運(yùn)行，send會(huì)把7這個(gè)值賦值給res變量

6.由于send方法中包含next()方法，所以程序會(huì)繼續(xù)向下運(yùn)行執(zhí)行print方法，然后再次進(jìn)入while循環(huán)

7.程序執(zhí)行再次遇到y(tǒng)ield關(guān)鍵字，yield會(huì)返回后面的值后，程序再次暫停，直到再次調(diào)用next方法或send方法。

這就結(jié)束了，說(shuō)一下，為什么用這個(gè)生成器，是因?yàn)槿绻肔ist的話，會(huì)占用更大的空間，比如說(shuō)取0,1,2,3,4,5,6............1000

你可能會(huì)這樣：

for n in range(1000):
    a=n

這個(gè)時(shí)候range(1000)就默認(rèn)生成一個(gè)含有1000個(gè)數(shù)的list了，所以很占內(nèi)存。

這個(gè)時(shí)候你可以用剛才的yield組合成生成器進(jìn)行實(shí)現(xiàn)，也可以用xrange(1000)這個(gè)生成器實(shí)現(xiàn)

yield組合：

def foo(num):
    print("starting...")
    while num<10:
        num=num+1
        yield num
for n in foo(0):
    print(n)

輸出：

starting...12345678910

xrange(1000):

for n in xrange(1000):    a=n

其中要注意的是python3時(shí)已經(jīng)沒(méi)有xrange()了，在python3中，range()就是xrange()了，你可以在python3中查看range()的類(lèi)型，它已經(jīng)是個(gè)<class 'range'>了，而不是一個(gè)list了，畢竟這個(gè)是需要優(yōu)化的。