自学python_14 生成器

生成器的介绍

推导式在应用时如果推导的元素过大时（比如说一个长度为千万的列表）如果只使用前几个推到的元素的话会浪费大量的空间，所以提出生成器的概念，可以将推到的结果逐个的获取，更加方便遍历。

生成器的使用

方法一：调用生成器自身的函数__next__。

list1 = [1,2,2,3,7]
g = (x + 50 for x in list1 if x < 7)
print(type(g))
print(g.__next__())
print(g.__next__())
print(g.__next__())
# <class 'generator'>
# 51
# 52
# 52

方法二：调用 builtins 系统内置函数 next 。

list1 = [1,2,2,3,7]
g = (x + 50 for x in list1 if x < 7)
print(type(g))
print(next(g))
print(next(g))
print(next(g))
# 51
# 52
# 52

每调用一次next就会产生一个元素，如果生成器将所有元素都生成一遍之后，就会抛出 StopIteration 异常。
用 try 和 except 改进的代码。

list1 = [1,2,2,3,7]
g = (x + 50 for x in list1 if x < 7)
while True:
    try:
        print(g.__next__())
    except StopIteration:
        print("没有更多元素！")
        break

函数变成生成器：
步骤：
1.定义一个函数，函数中使用yield关键字。
2.调用函数，接收调用的结果。
3.得到的结果就是生成器。
4.借助 next ,和 __next__得到想要的元素部分。

def func():
    n = 0
    while True:
        n += 1
        yield n
g = func()
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
# 1
# 2
# 3
# 4
# 5

这里的 yield 起到的是return 加暂停的作用，每次 yield 之后 n 的数值是不变的，再次调用只是继续循环而已。

利用生成器函数完成斐波那契数列：

def fib(length):
    a,b = 0,1
    n = 0
    while n < length:
        yield b
        a,b = b,a + b
        n += 1
    return "生成器函数没有东西能产生了"
g = fib(8)
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))
# StopIteration: 生成器函数没有东西能产生了
# 1 1 2 3 5 8 13 21 

在把fib数列前8项生成之后就返回了一个 StopIteration 并且提示信息是返回的字符串 ，return 之后是没有东西可以生成的时候返回的提示信息。

生成器中的send方法：

send 方法可以向每次生成器调用中传值。
注意：第一次调用必须要传None。

def func():
    i = 0
    while i < 5:
        temp = yield i
        print(temp)
        i += 1
    return "生成器函数没有东西能产生了"
g = func()
print(g.send(None))
print(g.send("1111111"))
print(g.send("2222222"))
print(g.send("3333333"))
# 0
# 1111111
# 1
# 2222222
# 2
# 3333333
# 3

我对 send 函数的理解是把每一次执行到 yield 那里会暂停返回，再进入程序时会继续往下执行。 send 就是在回到程序时把一个参数传递进来。

生成器的应用：协程

协程的底层是拿生成器完成的。

def task1(n):
    for i in range(n):
        print("正在搬第{}块砖".format(i))
def task2(n):
    for i in range(n):
        print("正在听{}首歌".format(i))

task1(10)
task2(5)

这样做只能把全部的砖搬完之后才会听歌，如果需要一种方法可以把听歌穿插在搬砖之间，使它们交替进行，这时就需要用到生成器。

def task1(n):
    for i in range(n):
        print("正在搬第{}块砖".format(i))
        yield None

def task2(n):
    for i in range(n):
        print("正在听{}首歌".format(i))
        yield None

g1 = task1(10)
g2 = task2(5)
while True:
    try:
        g1.__next__()
        g2.__next__()
    except:
        break
# 正在搬第0块砖
# 正在听0首歌
# 正在搬第1块砖
# 正在听1首歌
# 正在搬第2块砖
# 正在听2首歌
# 正在搬第3块砖
# 正在听3首歌
# 正在搬第4块砖
# 正在听4首歌
# 正在搬第5块砖