Python3 生成器的练习

生成器的各种练习

 

生成器有两种创建方式:

1. 生成器函数

　　拥有yield关键字的即是生成器函数

　　生成器函数特点: 调用生成器函数直接返回生成器的引用， 不执行函数内部代码， 每取一个值就执行代码段，遇到yield返回所取值

　　生成器取值:

1. next(generator) or generator.__next()方法取值
2. generator.send(value) 取值， 用法与next相同，只是会把发送的值保存到上一个yield的变量中, 不能用在第一个和最后一个yield
3. for xxx in generator: 利用for循环遍历取值，取完为止
4. 数据类型的强制转换 如list(generator)

2.  生成器表达式

　　所有的生成器本身也是个迭代器，生成器表达式方式如下

　　g = ( value for value in Iterable if value ...) 

 

生成器的一些demo

demo1: 生成器的取值问题

def demo():
    for i in range(4):
        yield i

g = demo()

g1 = (i for i in g)
g2 = (i for i in g1)

print(list(g1))
print(list(g2))

[0, 1, 2, 3]
[]

理解: 
# g, g1, g2 并没有开始取值，只是一个生成器，
# 当执行完第一个类型强制转换后， g1生成器内部已经没有值了, 
# 所以执行第二个类型转换, 'i for in g1' 这个操作时，遍历的是一个空列表,g2生成器转换的是一个控列表

 

demo2：生成器的取值问题2

def add(n, i):
    return n + i


def test():
    for i in range(4):
        yield i


g = test()
for n in [1, 10]:
    g = (add(n, i) for i in g)

print(list(g))

g = test()
for n in [1, 10]:
    g = (add(n, i) for i in g)
    g = list(g)

print(list(g))

[20, 21, 22, 23]
[11, 12, 13, 14]

理解:
"""
第一处可以把代码分开理解
g = test()
n = 1
g = (add(n, 1) for i in g) ->>> 此处g并未被取值
n = 10 
g = (add(n, 1) for i in g) ->>> 可以翻译成 
g = (add(n, 1) for i in (add(n, 1) for i in test()) ->>> test是初始的生成器,此时生成器均未被取值 
当执行list(g)时才开始取值 此时n=10 所以结果是 [20, 21, 22, 23]

第二处便好理解了, 每循环一次, 生成器便开始强制转换取值, 最后便是 [11, 12, 13, 14]
"""

 

 demo3： 生成器的预激活, send方法取值

# coding=utf-8

"""
功能：每次向一个容器里扔一个数, 返回这些数的最小值, 最大值, 平均值
要求： 用生成器实现
"""


def init(func):
    def inner(*args, **kwargs):
        g = func(*args, **kwargs)
        next(g)
        return g
    return inner


@init
def statistics():
    count = sum = avg = 0
    min = None
    max = None
    while True:
        data = yield {'min': min, 'max': max, 'avg': avg}
        if min is None:
            min = data
            max = data
            avg = data
            count = 1
            sum = data
        else:
            if data < min:
                min = data
            if data > min:
                max = data
            sum += data
            count += 1
            avg = sum / count


if __name__ == '__main__':
    s = statistics()
    print(s.send(-10))
    print(s.send(-5))
    print(s.send(30))

{'min': -10, 'avg': -10, 'max': -10}
{'min': -10, 'avg': -7.5, 'max': -5}
{'min': -10, 'avg': 5.0, 'max': 30}

 

 

 参考:  

　python之路——迭代器和生成器