Python生成器

生成器

• 在Python中，这种一边循环一边计算的机制，称为生成器：generator
• 生成器是一种特殊的迭代器

创建生成器方法1

• 列表生成式的[] 变为 （）

g = (x * 2 for x in range(5))
print(type(g))

运行结果:

创建生成器方法2

• 函数内添加yield

def gen():
    i = 0
    while i < 5:
        yield i
        i += 1


g = gen()
print(g)
print(next(g))
print(next(g))
print(next(g))
print(next(g))
print(next(g))

运行结果：

只要在def中有yield关键字的 就称为 生成器,代码中的gen()得到的是一个生成器，可以用next()来获取生成器的数据，生成器是可以迭代的

斐波那契数列

def fib(n):
    num1, num2 = 0, 1
    count = 0
    while count < n:
        num = num1
        num1, num2 = num2, num1 + num2
        count += 1
        yield num


f = fib(10)

print(next(f))
print(next(f))
print(next(f))
print(next(f))
print(next(f))
print(next(f))
print(next(f))
print(next(f))
print(next(f))

运行结果：

获取return的值

def fib(n):
    num1, num2 = 0, 1
    count = 0
    while count < n:
        num = num1
        num1, num2 = num2, num1 + num2
        count += 1
        yield num
    return '获取完毕'


f = fib(5)
print(next(f))
print(next(f))
print(next(f))
print(next(f))
print(next(f))

try:
    print(next(f))
except StopIteration as e:
    print('生成器返回值:%s'%e.value)

运行结果:

• return的值只有通过next() 获取越界，使用try捕获StopIteration错误，返回值就在Stoplteration的value中

总结yield next调用执行顺序

• 使用了yield关键字的函数不再是函数，而是生成器。
• 使用next调用生成器，执行顺序：
  • 如果是第一次执行，则从def代码块的开始部分执行，直到遇到yield为止，并且把yield关键字后的数值返回，当做next()的返回值, yield在这里有了return返回值的作用。
  • 如果不是第一次执行，则从上一次暂停的位置执行(即从上一次yield关键字的下一个语句开始执行)，直到遇到下一次yield为止，，并且把yield关键字后的数值返回，当做next()的返回值

使用send唤醒

• 使用send()函数的一个好处是可以在唤醒的同时向断点处传入一个附加数据。
• send不能在第一次发送非空内容
• 生成器还没有走到第一个yield语句，如果我们发生一个真实的值，这时是没有人去“接收”它的

def gen():
    i = 0
    while i < 5:
        temp = yield i
        print(temp)
        i += 1


f = gen()
print(next(f))
print('------------------------------')
print(f.send('古月'))
print('--------------------------------')
print(next(f))
print('-------------------------------')
print(f.send('唐舞麟'))

运行结果：

注意：

• 1.执行到yield时，gen函数作用暂时保存，返回i的值;
• 2.temp的值是下次f.send(“古月”)执行时获取的send发送过来的值，就是古月,
• 3.next(f)等价f.send(None),temp的值就是None

总结

• 使用了yield关键字的函数不再是函数，而是生成器
• yield关键字有两点作用：
  1. 保存当前运行状态（断点），然后暂停执行，即将生成器（函数）挂起
  2. 将yield关键字后面表达式的值作为返回值返回，此时可以理解为起到了return的作用
• 可以使用next()函数让生成器从断点处继续执行，即唤醒生成器（函数）
• Python3中的生成器可以使用return返回最终运行的返回值
• 生成器是这样一个函数，它记住上一次返回时在函数体中的位置。对生成器函数的第二次（或第 n 次）调用跳转至该函数中间，而上次调用的所有局部变量都保持不变。
• 生成器不仅“记住”了它数据状态；生成器还“记住”了它在流控制构造（在命令式编程中，这种构造不只是数据值）中的位置。
• 生成器的特点：存储的是生成数据的方式（即算法），而不是存储生成的数据，因此节约内存