淺談Python生成器generator之next和send的運行流程(詳解)
對于普通的生成器,第一個next調(diào)用,相當(dāng)于啟動生成器,會從生成器函數(shù)的第一行代碼開始執(zhí)行,直到第一次執(zhí)行完yield語句(第4行)后,跳出生成器函數(shù)。
然后第二個next調(diào)用,進入生成器函數(shù)后,從yield語句的下一句語句(第5行)開始執(zhí)行,然后重新運行到y(tǒng)ield語句,執(zhí)行后,跳出生成器函數(shù),后面再次調(diào)用next,依次類推。
下面是一個列子:
def consumer():
r = 'here'
for i in xrange(3):
yield r
r = '200 OK'+ str(i)
c = consumer()
n1 = c.next()
n2 = c.next()
n3 = c.next()
了解了next()如何讓包含yield的函數(shù)執(zhí)行后,我們再來看另外一個非常重要的函數(shù)send(msg)。其實next()和send()在一定意義上作用是相似的,區(qū)別是send()可以傳遞yield表達式的值進去,而next()不能傳遞特定的值,只能傳遞None進去。因此,我們可以看做c.next() 和 c.send(None) 作用是一樣的。
需要提醒的是,第一次調(diào)用時,請使用next()語句或是send(None),不能使用send發(fā)送一個非None的值,否則會出錯的,因為沒有Python yield語句來接收這個值。
下面來著重說明下send執(zhí)行的順序。當(dāng)?shù)谝淮蝧end(None)(對應(yīng)11行)時,啟動生成器,從生成器函數(shù)的第一行代碼開始執(zhí)行,直到第一次執(zhí)行完yield(對應(yīng)第4行)后,跳出生成器函數(shù)。這個過程中,n1一直沒有定義。
下面運行到send(1)時,進入生成器函數(shù),注意這里與調(diào)用next的不同。這里是從第4行開始執(zhí)行,把1賦值給n1,但是并不執(zhí)行yield部分。下面繼續(xù)從yield的下一語句繼續(xù)執(zhí)行,然后重新運行到y(tǒng)ield語句,執(zhí)行后,跳出生成器函數(shù)。
即send和next相比,只是開始多了一次賦值的動作,其他運行流程是相同的。
def consumer():
r = 'here'
while True:
n1 = yield r
if not n1:
return
print('[CONSUMER] Consuming %s...' % n1)
r = '200 OK'+str(n1)
def produce(c):
aa = c.send(None)
n = 0
while n < 5:
n = n + 1
print('[PRODUCER] Producing %s...' % n)
r1 = c.send(n)
print('[PRODUCER] Consumer return: %s' % r1)
c.close()
c = consumer()
produce(c)
運行結(jié)果:
[PRODUCER] Producing 1... [CONSUMER] Consuming 1... [PRODUCER] Consumer return: 200 OK1 [PRODUCER] Producing 2... [CONSUMER] Consuming 2... [PRODUCER] Consumer return: 200 OK2 [PRODUCER] Producing 3... [CONSUMER] Consuming 3... [PRODUCER] Consumer return: 200 OK3 [PRODUCER] Producing 4... [CONSUMER] Consuming 4... [PRODUCER] Consumer return: 200 OK4 [PRODUCER] Producing 5... [CONSUMER] Consuming 5... [PRODUCER] Consumer return: 200 OK5
以上這篇淺談Python生成器generator之next和send的運行流程(詳解)就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

