Python生成器 generator

生成器    只有在调用时才会生成相应的数据，只记录当前位置

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

或者你会写成这样：

a=[]
for i in range(10):
  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()