python中生成器是迭代器的一種筋粗,使用yield返回函數(shù)值蚂蕴。每次調(diào)用yield會暫停隧哮,而可以使用next()函數(shù)和send()函數(shù)可以恢復(fù)生成器羽历。
next函數(shù)與send函數(shù)很相似,都能獲得生成器的下一個yield后面表達式的值港令,不同的是send函數(shù)可以向生成器傳參啥容。
按照我的理解锈颗,可以把yield的功效理解為暫停和播放。
在一個函數(shù)中咪惠,程序執(zhí)行到y(tǒng)ield語句的時候击吱,程序暫停,返回yield后面表達式的值遥昧,在下一次調(diào)用的時候覆醇,從yield語句暫停的地方繼續(xù)執(zhí)行,如此循環(huán)炭臭,直到函數(shù)執(zhí)行完永脓。
注意到y(tǒng)ield是個表達式而不僅僅是個語句,所以可以使用x = yield r?這樣的語法徽缚。
這個知識點在協(xié)程中需要使用憨奸。協(xié)程的概念指的是在一個線程內(nèi),一個程序中斷去執(zhí)行另一個程序凿试,有點類似于CPU中斷排宰。這樣減少了切換線程帶來的負擔(dān),同時不需要多線程中的鎖機制那婉,因為不存在同時寫的問題板甘。
python使用生成器來實現(xiàn)協(xié)程,下面看一個python協(xié)程應(yīng)用于生產(chǎn)者消費者問題的例子
-----------------------------------------------------------------------------------------------------
?send(msg) 與 next()?
了解了next()如何讓包含yield的函數(shù)執(zhí)行后详炬,我們再來看另外一個非常重要的函數(shù)send(msg)盐类。其實next()和send()在一定意義上作用是相似的,區(qū)別是send()可以傳遞yield表達式的值進去呛谜,而next()不能傳遞特定的值在跳,只能傳遞None進去。因此隐岛,我們可以看做c.next()(版本next(c)) 和 c.send(None) 作用是一樣的猫妙。來看這個例子:
表達式被賦予了’Fighting!’輸出的結(jié)果為:?
Wen Chuan Fighting!?
需要提醒的是,第一次調(diào)用時聚凹,請使用next()語句或是send(None)割坠,不能使用send發(fā)送一個非None的值,否則會出錯的妒牙,因為沒有Python yield語句來接收這個值彼哼。?
-----------------------------------------------------------------------------------------------------
send(msg) 與 next()的返回值
send(msg) 和 next()是有返回值的,它們的返回值很特殊湘今,返回的是下一個yield表達式的參數(shù)敢朱。比如yield 5,則返回 5 。到這里蔫饰,是不是明白了一些什么東西琅豆?本文第一個例子中愉豺,通過for i in alist 遍歷 Generator篓吁,其實是每次都調(diào)用了alist.Next(),而每次alist.Next()的返回值正是yield的參數(shù)蚪拦,即我們開始認為被壓進去的東東杖剪。我們再延續(xù)上面的例子:
throw()和close()中斷 Generator?
中斷Generator是一個非常靈活的技巧,可以通過throw拋出一個GeneratorExit異常來終止Generator驰贷。Close()方法作用是一樣的盛嘿,其實內(nèi)部它是調(diào)用了throw(GeneratorExit)的。我們看:
def close(self):?
try:?
self.throw(GeneratorExit)?
except (GeneratorExit, StopIteration):?
pass?
else:?
raise RuntimeError(“generator ignored GeneratorExit”)?
Other exceptions are not caught?
因此括袒,當我們調(diào)用了close()方法后次兆,再調(diào)用next()或是send(msg)的話會拋出一個異常:
Traceback (most recent call last):?
File “/home/evergreen/Codes/yidld.py”, line 14, in?
d = c.send(‘Fighting!’) #d 獲取了yield 12 的參數(shù)值12?
StopIteration