1. 程式人生 > >Python生成器generator之next和send運行流程

Python生成器generator之next和send運行流程

表達式 運行流程 其他 ... 一行 xrange 沒有 否則 重要

對於普通的生成器,第一個next調用,相當於啟動生成器,會從生成器函數的第一行代碼開始執行,直到第一次執行完yield語句(第4行)後,跳出生成器函數。

然後第二個next調用,進入生成器函數後,從yield語句的下一句語句(第5行)開始執行,然後重新運行到yield語句,執行後,跳出生成器函數,

後面再次調用next,依次類推。下面是一個列子:

1 def consumer():
2 r = ‘here‘
3 for i in xrange(3):
4 yield r
5 r = ‘200 OK‘+ str(i)
6
7 c = consumer()
8 n1 = c.next()

9 n2 = c.next()
10 n3 = c.next()

了解了next()如何讓包含yield的函數執行後,我們再來看另外一個非常重要的函數send(msg)。其實next()和send()在一定意義上作用是相似的,區別是send()可以傳遞yield表達式的值進去,而next()不能傳遞特定的值,只能傳遞None進去。因此,我們可以看做c.next() 和 c.send(None) 作用是一樣的。需要提醒的是,第一次調用時,請使用next()語句或是send(None),不能使用send發送一個非None的值,否則會出錯的,因為沒有Python yield語句來接收這個值。
下面來著重說明下send執行的順序。當第一次send(None)(對應11行)時,啟動生成器,從生成器函數的第一行代碼開始執行,直到第一次執行完yield(對應第4行)後,跳出生成器函數。這個過程中,n1一直沒有定義。

下面運行到send(1)時,進入生成器函數,註意這裏與調用next的不同。這裏是從第4行開始執行,把1賦值給n1,但是並不執行yield部分。下面繼續從yield的下一語句繼續執行,然後重新運行到yield語句,執行後,跳出生成器函數。

即send和next相比,只是開始多了一次賦值的動作,其他運行流程是相同的。

1 def consumer():
2 r = ‘here‘
3 while True:
4 n1 = yield r
5 if not n1:
6 return
7 print(‘[CONSUMER] Consuming %s...‘ % n1)

8 r = ‘200 OK‘+str(n1)
9
10 def produce(c):
11 aa = c.send(None)
12 n = 0
13 while n < 5:
14 n = n + 1
15 print(‘[PRODUCER] Producing %s...‘ % n)
16 r1 = c.send(n)
17 print(‘[PRODUCER] Consumer return: %s‘ % r1)
18 c.close()
19
20 c = consumer()
21 produce(c)

運行結果:

[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運行流程