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

原文地址：https://blog.csdn.net/pfm685757/article/details/49924099

 

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

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

後面再次調用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的函數執行後，我們再來看另外一個非常重要的函數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相比，只是開始多了一次賦值的動作，其他運行流程是相同的。

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)

運行結果：

[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