关于python生成器generator的简单使用以及next到头继续返回值的方法
一、generator的简单描述
python的生成器generator是一个非常科学读取数据方法。它的值返回需要用到关键字yield. 有点类似return的效果,不同的是yield会不结束与释放整个函数,相当于指针停留在这个地方,等待下次来取的时候继续向下读取,而return会立即结束整个函数与内存,得到最终一个变量结果。一般我们在循环读取数据时,其实是进行重复的函数打开->释放->得到值,打开->释放->得到值。但是对于生成器来说,它只需要打开一次,取一次值然后停留这个值的位置,然后等下次来取的时候再读取下一个值,也就是读取->停留->下一个。
二、generator的值的读取方法
使用next()函数可以主动读取生成器的返回值。每次调用返下一个值。需要注意的是,如果next到头了值读完了再继续读取的化就会报错。但是我们在运行程序的时候,有时候希望它到头了能返回一个我们指定的一个值。怎么办呢?可以使用next(generator,default)方法来设置默认值
示例代码:
def test():
i=0
while i<5:
yield i
print('temp:',i)
i+=1
if i>=5:
yield 5
print('下面没有了',i)
break
for i in range(10):
v=next(a,5)
print(v)
结果如下:
0
temp: 0
1
temp: 1
2
temp: 2
3
temp: 3
4
temp: 4
5
下面没有了 5
5
5
5
5