一日一技：生成器里面的 return 有什么用？

在粉絲群里，有一位同學(xué)問(wèn)了這樣一個(gè)問(wèn)題：

def gen_data(num): if num > 10: for i in range(num): yield i else: return num generator = gen_data(5) for num in generator: print(num)

當(dāng)傳入的的參數(shù)小于等于10 的時(shí)候，為什么沒(méi)有返回這個(gè)參數(shù)本身？

這道題，當(dāng)我們傳入的參數(shù)大于10的時(shí)候，能得到符合預(yù)期的結(jié)果，如下圖所示：

但是，當(dāng)我們傳入數(shù)據(jù)5的時(shí)候，我們來(lái)看看運(yùn)行效果：

可以看到，數(shù)字5并沒(méi)有被打印出來(lái)，程序直接運(yùn)行到了最后。

之所以會(huì)出現(xiàn)這種情況，是因?yàn)檫@個(gè)同學(xué)以為，當(dāng)參數(shù)大于10的時(shí)候，gen_data(12)返回的是生成器，而當(dāng)參數(shù)不大于10的時(shí)候，返回的是一個(gè)數(shù)字。顯然這樣的想法是不對(duì)的，否則，for num in 10這種語(yǔ)法早就報(bào)錯(cuò)了，數(shù)字是不能被迭代的。

正確的說(shuō)法應(yīng)該是，因?yàn)間en_data里面有yield，所以gen_data(參數(shù))返回一個(gè)生成器。無(wú)論參數(shù)傳入的是什么，返回的都是生成器。如下圖所示：

為了說(shuō)明為什么傳入?yún)?shù)為5的時(shí)候，for 循環(huán)不執(zhí)行，我們簡(jiǎn)化一下代碼：

def gen_data(): yield 1 yield 2 yield 3 return 4 generator = gen_data() for num in generator: print(num)

運(yùn)行效果如下圖所示：

可以看到，對(duì)于這樣一個(gè)非常簡(jiǎn)單的生成器，在 for 循環(huán)里面也只是打印了數(shù)字123，并沒(méi)有打印數(shù)字4。

關(guān)于生成器中的return，我們可以從 Python 官方文檔PEP 255 — Simple Generators[1]中找到說(shuō)明：

return?在生成器中，表示生成器運(yùn)行完成了，可以結(jié)束了。然后生成器會(huì)拋出一個(gè)StopIteration的異常。而for循環(huán)能夠檢測(cè)到這個(gè)異常，于是結(jié)束循環(huán)。所以當(dāng)我們傳入的參數(shù)為5的時(shí)候，生成器直接運(yùn)行到了?return，于是它直接就拋出StopIteration，于是 for 循環(huán)檢測(cè)到這個(gè)異常就結(jié)束了。

在生成器里面的return只是一個(gè)結(jié)束標(biāo)志，它不會(huì)把后面寫的值返回給調(diào)用者。這跟函數(shù)里面的return語(yǔ)句是不一樣的。

參考文獻(xiàn)

Python

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實(shí)的內(nèi)容，請(qǐng)聯(lián)系我們jiasou666@gmail.com 處理，核實(shí)后本網(wǎng)站將在24小時(shí)內(nèi)刪除侵權(quán)內(nèi)容。

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實(shí)的內(nèi)容，請(qǐng)聯(lián)系我們jiasou666@gmail.com 處理，核實(shí)后本網(wǎng)站將在24小時(shí)內(nèi)刪除侵權(quán)內(nèi)容。