python進階（11）生成器

生成器

利用迭代器，咱們能夠在每次迭代獲取數據（經過next()方法）時按照特定的規律進行生成。可是咱們在實現一個迭代器時，關於當前迭代到的狀態須要咱們本身記錄，進而才能根據當前狀態生成下一個數據。
爲了達到記錄當前狀態，並配合next()函數進行迭代使用，咱們能夠採用更簡便的語法，即生成器(generator)。生成器是一類特殊的迭代器。
 

建立生成器方法1

要建立一個生成器，有不少種方法。第一種方法很簡單，只要把一個列表生成式的 [ ] 改爲 ( )

>>> L = [x**2 for x in range(5)]
>>> L
[0, 1, 4, 9, 16]
>>> G = (x**2 for x in range(5))
>>> G
<generator object <genexpr> at 0x7fb63d218750>

建立 L 和 G 的區別僅在於最外層的 [ ] 和 ( ) ， L 是一個列表，而 G 是一個生成器。咱們能夠直接打印出列表L的每個元素，而對於生成器G，咱們能夠按照迭代器的使用方法來使用，便可以經過next()函數、for循環、list()等方法使用。

>>> next(G)
0
>>> next(G)
1
>>> next(G)
4
>>> next(G)
9
>>> next(G)
16
>>> next(G)
Traceback (most recent call last):
  File "<pyshell#39>", line 1, in <module>
    next(G)
StopIteration

>>> G = ( x**2 for x in range(5))
>>> for x in G:
	print(x)

 

建立生成器方法2

generator很是強大。若是推算的算法比較複雜，用相似列表生成式的 for 循環沒法實現的時候，還能夠用函數來實現。
就像以前提到的斐波那切數列
 

注意，在用迭代器實現的方式中，咱們要藉助幾個變量(n、current、num一、num2)來保存迭代的狀態。如今咱們用生成器來實現一下。

>>> def fib(n):
	current = 0 
	num1, num2 = 0, 1
	while current < n:
		num = num1
		num1, num2 = num2, num1 + num2
		current += 1
		yield num
	return '完成'

>>> F = fib(5)
>>> next(F)
0
>>> next(F)
1
>>> next(F)
1
>>> next(F)
2
>>> next(F)
3
>>> next(F)
Traceback (most recent call last):
  File "<pyshell#80>", line 1, in <module>
    next(F)
StopIteration: 完成

在使用生成器實現的方式中，咱們將本來在迭代器__next__方法中實現的基本邏輯放到一個函數中來實現，可是將每次迭代返回數值的return換成了yield，此時新定義的函數便再也不是函數，而是一個生成器了。
簡單來講：只要在def中有yield關鍵字的 就稱爲 生成器
 
此時按照調用函數的方式( 案例中爲F = fib(5) )使用生成器就再也不是執行函數體了，而是會返回一個生成器對象（ 案例中爲F ），而後就能夠按照使用迭代器的方式來使用生成器了。

>>> for n in fib(5):
	print(n)

0
1
1
2
3
>>>

可是用for循環調用generator時，發現拿不到generator的return語句的返回值。若是想要拿到返回值，必須捕獲StopIteration錯誤，返回值包含在StopIteration的value中：

>>> g = fib(5)
>>> while True:
	try:
		x = next(g)
		print(f"value:{x}")
	except StopIteration as e:
		print(f"生成器返回值:{e.value}")
		break

value:0
value:1
value:1
value:2
value:3
生成器返回值:完成

 

總結

• 使用了yield關鍵字的函數再也不是函數，而是生成器。（使用了yield的函數就是生成器）
• yield關鍵字有兩點做用：
  • 保存當前運行狀態（斷點），而後暫停執行，即將生成器（函數）掛起
  • 將yield關鍵字後面表達式的值做爲返回值返回，此時能夠理解爲起到了return的做用
• 可使用next()函數讓生成器從斷點處繼續執行，即喚醒生成器（函數）
• Python3中的生成器可使用return返回最終運行的返回值，而Python2中的生成器不容許使用return返回一個返回值（便可以使用return從生成器中退出，但return後不能有任何表達式）。
   

使用send喚醒

咱們除了可使用next()函數來喚醒生成器繼續執行外，還可使用send()函數來喚醒執行。使用send()函數的一個好處是能夠在喚醒的同時向斷點處傳入一個附加數據。
 
例子：執行到yield時，gen函數做用暫時保存，返回i的值; temp接收下次c.send("python")，send發送過來的值，c.next()等價c.send(None)

>>> def gen():
	i = 0
	while i < 5:
		temp = yield i
		print(temp)
		i += 1

使用send

>>> f = gen()
>>> next(f)
0
>>> f.send('haha')
haha
1
>>> next(f)
None
2
>>> f.send('haha')
haha
3
>>>