生成器(generator)
生成器是一种返回生成序列的方法,与直接使用列表等方式返回序列的方式不同的是,他的生成可以是无限的。
生成器可以与next搭配使用,可以被看作是一种特殊的迭代器。
yield语句
yield一般与循环相配合使用,我们以斐波那契数列为例构建生成器
注:斐波那契数列是一个特殊的增数列,后一个数是前两个数之和,斐波那契数列如下:1,1,2,3,5, 8,…
def fibonacci():
a, b = 0, 1
while True:
yield b
a, b = b, a + b
在代码中展示了一个在死循环中生成斐波那契数列的功能,使用yield进行返回。使用next可以获取下一个斐波那契数列的数字。
组合使用生成器
不同的生成器可以组合使用,但这种用法一般要求数列有限。
下面我们创建一个计算平方的函数和一个值+1的函数
def power(values):
for value in values:
yield value ** 2
def add(values):
for value in values:
yield value + 1
下面我们使用前面介绍过的列表推导生成方式生成一个由数字组成的序列
values = [i for i in range(5)]
生成器单个使用效果如下:
合并使用效果如下
这样的结果就是先+1再平方