python的generator生成器用法测试

yield、send、threw、close

# coding: utf8# 生成器
def gen(n):for i in range(n):yield ig = gen(5)      # 创建一个生成器
print(g)        # <generator object gen at 0x10bb46f50>
print(type(g))  # <type 'generator'># 迭代生成器中的数据(只有执行for循环的时候，gen生成器才会返回yield后面的值，而且是一个一个的返回，这不同于return，全部返回来)
for i in g:print(i)#0 1 2 3 4'''
next
generator.__next__()：执行 for 时调用此方法，每次执行到 yield 就会停止，然后返回 yield 后面的值，
如果没有数据可迭代，抛出 StopIterator 异常，for 循环结束
'''
def gen1(n):for i in range(n):print('start')yield iprint('end')g = gen1(3)      # 创建一个生成器
print(g.__next__())  # 0
print(g.__next__())  # 1
print(g.__next__())  # 2
print(g.__next__())  # StopIteration'''
send (send 方法可以把外部的值传入生成器内部，从而改变生成器的状态)
当我们执行 g.send(-1) 时，相当于把 -1 传入到了生成器内部，然后赋值给了 yield 前面的 j，
此时 j = -1，然后这个方法就会 break 出来，不会继续迭代下去。
'''
def gen2():i = 1while True:j = yield ii *= 2if j == -1:breakg = gen2()   # 创建一个生成器
print(g.__next__())  # 1
print(g.__next__())  # 2
print(g.__next__())  # 4
# send 把 -1 传入生成器内部 走到了 j = -1 这个分支
print(g.send(-1))   # StopIteration 迭代停止'''
throw(外部除了可以向生成器内部传入一个值外，还可以传入一个异常，也就是调用 throw 方法)
'''
def gen3():try:yield 1except ValueError:yield 'ValueError'finally:print('finally')g = gen3()   # 创建一个生成器
print(g.__next__()) # 1
# 向生成器内部传入异常 返回ValueError
print(g.throw(ValueError))# Output：
# 1
# ValueError
# finally'''
close(生成器的 close 方法也比较简单，就是手动关闭这个生成器，关闭后的生成器无法再进行操作)
'''
g = gen()
print(g.close()) # 关闭生成器
print(g.__next__()) # 无法迭代数据

利用生成器做一个生产和消费协同并发：

# coding: utf8def consumer():i = Nonewhile True:# 拿到 producer 发来的数据j = yield iprint('consume %s' % j)def producer(c):c.__next__()for i in range(5):print('produce %s' % i)# 发数据给 consumerc.send(i)c.close()c = consumer()
producer(c)