首页 > 代码库 > python--生成器
python--生成器
生成器
一、概念
python中的生成器(generator)保存的是算法，只有当真正调用需要的值的时候才会去计算出值。它是一种惰性计算（lazy evaluation）。
二、列表生成式
定义一个列表
a = [0,1,2,3,4,5,6,7,8,9]
除了上面直接定义外，我们还可以使用列表生成式：
a = [i for i in range(10)] print(a) #输出 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
三、生成器
通过上面的列表生成式，我们可以直接创建一个列表。但是受到内存的限制，列表的容量是有限的。而且如果我们创建了一个包含数百万元素的列表，并且只需要用到前面几个元素，那就有大部分空间浪费了。
所以，如果列表元素可以按照某种算法推算出来，那我们是否可以在循环的过程中不断的推算后续的元素？这样就不必创建完整的list，从而节省大量的空间。在pyhong中，一边循环一边计算的机制，成为生成器：generator。
要创建一个generator有很多种方法，我们可以直接将列表生成式改成生成器：
>>> a = (i for i in range(10)) >>> a <generator object <genexpr> at 0x0000024897F6B4C0>
3.1 列表生成式和生成器
a = (i for i in range(100000000)) b = [i for i in range(100000000)]
a的生成时间是瞬时的，并且不会占用什么内存空间。因为生成器根本就没有生成所需的内容，只有在访问的时候才会生成相应的值
3.2 访问生成器
- 生成器只保留当前位置的值，直接访问 a[5]是没有值的（报错）
- 只有一个__next()__方法
- 只能一个一个取值，并且只能往后取值，不能返回取前面的值
使用循环来取值:
a = (i for i in range(10))
for i in a:
print(i)
#输出
0
1
2
3
4
5
6
7
8
9
使用__next()__方法取值:
a = (i for i in range(10))
print(a.__next__())
print(a.__next__())
print(a.__next__())
print("插入")
print(a.__next__())
print(a.__next__())
#输出
0
1
2
插入
3
4
每调用一个__next()__方法，我们就能取到一个值，并且我们只能往后取值。生成器能记住当前位置的值，但是并不知道前后位置的值，这也是生成器能够节省内存的原因。只有调用到该位置时才会生成相应的数据。
四、yield将函数转换成生成器
利用列表只能实现一部分的功能，一下复杂的运行就不能通过列表生成式这样的来实现了。但是我们可以将一个函数变成生成器。
斐波那契函数:
def fib(n):
a = 1
b = 1
number = 0
while number < n:
print(b)
a,b = b,a+b
number += 1
将该函数变成生成器只需要将print(b) 改为 yield b
def fib(n):
a = 0
b = 1
number = 0
while number < n:
#print(b)
yield b
a,b = b,a+b
number += 1
f = fib(10)
print(f.__next__())
print(f.__next__())
print(f.__next__())
print(f.__next__())
#输出
1
1
2
3
五、生成器结束
生成器不会把结果保存在一个系列中，而是保存生成器的状态，在每次进行迭代时返回一个值，直到遇到StopIteration异常结束。
def fib(n):
a = 0
b = 1
number = 0
while number < n:
#print(b)
yield b
a,b = b,a+b
number += 1
return ‘--done--‘
f = fib(6)
while True:
try:
x = next(f)
print(x)
except StopIteration as e:
print(‘Generator return value:‘,e.value)
break
#输出
1
1
2
3
5
8
Generator return value: --done--
python--生成器