09 生成器函数进阶

def generator():
	print('123')
	yield 1
	print('456')
	yield 2
	print('789')
g = generator()
ret = g.__next__()
print('*',ret)
ret = g.__next__()
print('**',ret)
ret = g.__next__()
print('***',ret)

输出：因为print(‘789’)下面没有yield了，所以会报错。

send的使用

def generator():
	print('123')
	content = yield 1
	print('^^^^^^^^',content)
	print('456')
	yield 2
g = generator()
ret = g.__next__()
print('***',ret)
ret = g.send('hello')
print('***',ret)

输出：

send获取下一个之的效果和next基本一致，只不过在获取下一个值的时候，给上一个yield的位置传第一个数据。

使用send的注意事项。

1. 第一次使用生成器的时候，使用next获取下一个值。
   
2. 最后一个yield不能接受外部的值。

例子：

获取移动平均值

def average():
    sum = 0
    count = 0
    avg = 0
    while True:
        num = yield avg
        sum += num    # 10
        count += 1    # 1
        avg = sum/count

avg_g = average()
avg_g.__next__()
avg1 = avg_g.send(10)
avg1 = avg_g.send(20)
print(avg1)

输出：

带有生成器功能的装饰器

def init(func):   #装饰器
    def inner(*args,**kwargs):
        g = func(*args,**kwargs)    #g = average()
        g.__next__()
        return g
    return inner

@init
def average():
    sum = 0
    count = 0
    avg = 0
    while True:
        num = yield avg
        sum += num    # 10
        count += 1    # 1
        avg = sum/count

avg_g = average()   #===> inner
ret = avg_g.send(10)
print(ret)
ret = avg_g.send(20)
print(ret)

输出：

yield from

def generator1():
    a = 'abcde'
    b = '12345'
    for i in a:
        yield i
    for i in b:
        yield i
g1 = generator1()
for i in g1:
    print(i)

输出：

def generator2():
    a = 'abcde'
    b = '12345'
    yield from a
    yield from b
g1 = generator2()
for i in g1:
    print(i)

输出：