Python的生成器send()方法 & yield_from

生成器對象是一個迭代器。可是它比迭代器對象多了一些方法，它們包括send方法，throw方法和close方法。這些方法，主要是用於外部與生成器對象的交互。本文先介紹send方法。

send

send方法有一個參數，該參數指定的是上一次被掛起的yield語句的返回值。這樣提及來比較抽象，看下面的例子。

def MyGenerator():
value = (yield 1)
value = (yield value)


gen = MyGenerator()
print （next(gen)）
print gen.send(2)
print gen.send(3)
輸出的結果以下
1
2
Traceback (most recent call last): File "test.py", line 18, in <module> print gen.send(3) StopIteration

 

上面代碼的運行過程以下。
當調用gen.next()方法時，python首先會執行MyGenerator方法的yield 1語句。因爲是一個yield語句，所以方法的執行過程被掛起，而next方法返回值爲yield關鍵字後面表達式的值，即爲1。

當調用gen.send(2)方法時，python首先恢復MyGenerator方法的運行環境。同時，將表達式(yield 1)的返回值定義爲send方法參數的值，即爲2。這樣，接下來value=（yield 1）這一賦值語句會將value的值置爲2。繼續運行會遇到yield value語句。所以，MyGenerator方法再次被掛起。同時，send方法的返回值爲yield關鍵字後面表達式的值，也即value的值，爲2。

當調用send(3)方法時MyGenerator方法的運行環境。同時，將表達式(yield value)的返回值定義爲send方法參數的值，即爲3。這樣，接下來value=（yield value）這一賦值語句會將value的值置爲3。繼續運行，MyGenerator方法執行完畢，故而拋出StopIteration異常。

注意一個要點：生成器函數在運行時候遇到yield時候就會掛起來，不會往下繼續運行，直到有人調用.next()或者send()纔會將值拋出而後往下運行。

總的來講，send方法和next方法惟一的區別是在執行send方法會首先把上一次掛起的yield語句的返回值經過參數設定，從而實現與生成器方法的交互。可是須要注意，在一個生成器對象沒有執行next方法以前，因爲沒有yield語句被掛起，因此執行send方法會報錯。例如

gen = MyGenerator()
print gen.send(2)

上面代碼的輸出爲
Traceback (most recent call last):
File "test.py", line 16, in <module>
print gen.send(2)
TypeError: can't send non-None value to a just-started generator

固然，下面的代碼是能夠接受的

gen = MyGenerator()
print gen.send(None)

 

由於當send方法的參數爲None時，它與next方法徹底等價。可是注意，雖然上面的代碼能夠接受，可是不規範。因此，在調用send方法以前，仍是先調用一次next方法爲好。

yield from （功能十分強大）

　　yield from 後面必須接一個iterable對象。

def gen_list(list):
    yield from list

d = gen_list(['frank', 'yangchao', 'liuliu'])
for i in d:
    print(i)

frank
yangchao
liuliu

　　yield與yield from()的區別以下：yield from 能夠將可迭代對象中的每一個對象迭代出來。

def yield_func(iterable):
    yield iterable
def yield_from_func(iterable):
    yield from iterable

yield_gen = yield_func(range(10))
for i in yield_gen:
    print(i)
yield_from_gen = yield_from_func(range(10))
for i in yield_from_gen:
    print(i)

range(0, 10)
0
1
2
3
4
5
6
7
8
9