1. 程式人生 > >Python 生成器 generator

Python 生成器 generator

調用 href end term AR none png -o 賦值

前提條件: 如果函數含有yield就是生成器,執行過程中遇到yield就跳出。

舉例:
def gen():
for i in range(10):
x = yield i
print(x)
g=ge()
print(g.send(None)))
print(g.send(2))
技術分享圖片
先說表達式x = yield i
如果這個表達式只是x = i, 相信每個人都能理解。即把i的值賦值給了x,而現在等號右邊是一個yield i,所以先要執行yield i,然後才是賦值。
因為生成器遇到yield就跳出,所以yield把i值返回到了調用者那裏。
這個表達式的下一步操作:賦值。卻因為等號右邊的yield被暫停了,換句話說x = yield i才執行了一半,當調用者通過send(var)回到生成器函數時是回到之前那個賦值表達式被暫停的那裏,所以接下來執行x = yield i的另一半,那就是這個賦值操作啦,這個值正是調用者通過send(var)發送進生成器的值。

再舉例:
def consumer():
r = ‘‘
while True:
n = yield r
if not n:
return
print(‘[CONSUMER] Consuming %s...‘ % n)
r = ‘200 OK‘

def produce(c):
c.send(None)
n = 0
while n < 5:
n = n + 1
print(‘[PRODUCER] Producing %s...‘ % n)
r = c.send(n)
print(‘[PRODUCER] Consumer return: %s‘ % r)
c.close()

c = consumer()
produce(c)
執行結果:

[PRODUCER] Producing 1...
[CONSUMER] Consuming 1...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 2...
[CONSUMER] Consuming 2...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 3...
[CONSUMER] Consuming 3...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 4...
[CONSUMER] Consuming 4...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 5...
[CONSUMER] Consuming 5...
[PRODUCER] Consumer return: 200 OK

官方說明:
註意到consumer函數是一個generator,把一個consumer傳入produce後:
首先調用c.send(None)啟動生成器;然後,一旦生產了東西,通過c.send(n)切換到consumer執行;consumer通過yield拿到消息,處理,又通過yield把結果傳回;produce拿到consumer處理的結果,繼續生產下一條消息;produce決定不生產了,通過c.close()關閉consumer,整個過程結束。

我的理解:
請註意produce()函數的最開始的地方是c.send(None),正如上面所說這個操作是初始化generator,並做第一個yield(遇到consumer函數就返回),它是有值的,他的值r為‘‘,如果你執行命令行的話,你會發現結果先空了一行,就是他的功勞。 然後到n=0+1=1,此時執行c.send(1) ,它開始的位置是在給consumer函數中的n賦值,即[CONSUMER] Consuming 1 ,繼續執行命令,此時r被賦值為‘200 ok’ ,因為while True 會一直循環,所以繼續重來,但是生成器遇到yield 會自動跳出,此時跳出結果變成為r=‘200 ok’。
補充知識:
while True :
只有遇到continue 和break才能跳出循環。在生成器裏面遇到yield也能。
if not x:相當於if x is not Noneif not x is None`
在python中 None, False, 空字符串"", 0, 空列表[], 空字典{}, 空元組()都相當於False

參考地址如下:
python
廖雪峰

Python 生成器 generator