Python生成器 generator

生成器    只有在調用時才會生成相應的數據，只記錄當前位置

列表生成式
2 l = [x*2 for x in range(10)] 3 print(l)
執行結果： [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

或者你會寫成這樣：

a=[] 2 for i in range(10): 3   a.append(i*2)

 要創建一個generator，最簡單的就是，把一個列表生成式的 [] 改為 ()

g = (x * x for x in range(10))
for i in g:
  print(i)
執行結果：

0
2
4
6
8
10
12
14
16
18

generator非常強大，如果推算的算法比較復雜，用類似列表生成式的 for循環無法實現的時候，還可以用函數來實現。

比如，著名的斐波那契數列（Fibonacci），除數列從第3項開始，每一項都等於前兩項之和。

1，1，2，3，5，8，13，21，34，.....

def fib(max):
    n,a,b = 0,0,1
    while n<max:
        print(b)
        a, b = b,a+b
        n +=1  # n = n+1
    return 'done'
fib(8)
執行結果：

1
1
2
3
5
8
13
21

變成generator 只要把 print(b) 改為 yield b 即可。

def fib(max):
    n,a,b = 0,0,1
    while n<max:
        # print(b)
        yield b
        a, b = b,a+b
        n +=1
    return 'done'
f= fib(8)
for i in f:
    print(i)

 也可以捕獲異常

while True:
    try:
        x= next(g)
        print('g:',x)
    except StopIteration as e:
        print('Generator return value:',e.value)
        break

 generator並行，吃包子程序。

import time
def consumer(name):
    print('%s 准備吃包子啦！'%name)
    while True:
        baozi = yield
        print('[%s]包子來了，被[%s]吃了！'%(baozi,name))
'''
c = consumer('Jason')
c.__next__()
b1 = '鮮肉'
c.send(b1)
c.__next__()
'''
def producer():
    c1 = consumer('Tim')
    c2= consumer('Jack')
    c1.__next__()
    c2.__next__()
    print('開始做包子啦！！')
    for i in range(10):
        time.sleep(1)
        print('做了2個包子。')
        c1.send(i)
        c2.send(i)

producer()