之前一直對python的yield理解的有點(diǎn)模糊峻村,昨天搜索發(fā)現(xiàn)了一個(gè)講解非常清楚的版本锡凝,所以mark一下。
來源:python中yield的用法詳解——最簡單窜锯,最清晰的解釋
首先,如果你還沒有對yield有個(gè)初步分認(rèn)識吞瞪,那么你先把yield看做“return”芍秆,這個(gè)是直觀的,它首先是個(gè)return霉颠,普通的return是什么意思荆虱,就是在程序中返回某個(gè)值,返回之后程序就不再往下運(yùn)行了克伊。看做return之后再把它看做一個(gè)是生成器(generator)的一部分(帶yield的函數(shù)才是真正的迭代器)不从,好了犁跪,如果你對這些不明白的話,那先把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))
就這么簡單的幾行代碼就讓你明白什么是yield,代碼的輸出這個(gè):
starting...
4
********************
res: None
4
我直接解釋代碼運(yùn)行順序乏矾,相當(dāng)于代碼單步調(diào)試:
程序開始執(zhí)行以后迁杨,因?yàn)閒oo函數(shù)中有yield關(guān)鍵字,所以foo函數(shù)并不會(huì)真的執(zhí)行铅协,而是先得到一個(gè)生成器g(相當(dāng)于一個(gè)對象)
直到調(diào)用next方法,foo函數(shù)正式開始執(zhí)行痒给,先執(zhí)行foo函數(shù)中的print方法骏全,然后進(jìn)入while循環(huán)
程序遇到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é)果,
程序執(zhí)行print(""20)宾添,輸出20個(gè)*
又開始執(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,
程序會(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開始的地方是接著上一次的next停止的地方執(zhí)行的朴爬,所以調(diào)用next的時(shí)候橡淆,生成器并不會(huì)從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),程序會(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é)束了,說一下藕漱,為什么用這個(gè)生成器崭闲,是因?yàn)槿绻肔ist的話,會(huì)占用更大的空間刁俭,比如說取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...
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)化的哑了。
作者:馮爽朗
來源:CSDN
原文:https://blog.csdn.net/mieleizhi0522/article/details/82142856