python3 yield 理解

轉(zhuǎn)載文章來源:python中yield的用法詳解——最簡單,最清晰的解釋


首先我要吐槽一下,看程序的過程中遇見了yield這個(gè)關(guān)鍵字,然后百度的時(shí)候,發(fā)現(xiàn)沒有一個(gè)能簡單的讓我懂的,講起來真TM的都是頭頭是道,什么參數(shù),什么傳遞的,還口口聲聲說自己的教程是最簡單的,最淺顯易懂的,我就想問沒有有考慮過讀者的感受。

接下來是正題:

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

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

就這么簡單的幾行代碼就讓你明白什么是yield,代碼的輸出這個(gè):

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

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

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

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

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

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

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

6.程序會繼續(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開始的地方是接著上一次的next停止的地方執(zhí)行的,所以調(diào)用next的時(shí)候,生成器并不會從foo函數(shù)的開始執(zhí)行,只是接著上一步停止的地方開始,然后遇到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

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

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

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

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

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

你可能會這樣:

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...
1
2
3
4
5
6
7
8
9
10

xrange(1000):

for n in xrange(1000):
    a=n

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

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容