在Python中,一边循环一边计算的机制,称为生成器:Generator
1. 生成器的应用
性能限制时需要用到,比如在读取一个10G的文件时,如果一次性将10G的文件加载到内存处理的话(read方法),内存肯定会溢出;但使用生成器把读写交叉进行,如使用(readline和readlines)就可以在循环读取的同时不断处理,这样就可以节省大量的内存空间。
- 如:使用列表生成式时,会将打印好的内容一次性全部输出,如果我们写成
range(10000000)
等待时间就会很长,此处也能验证什么时候需要生成器?
这一点。
2. 生成器的创建方法
- 方法1 :列表生成式的改写,[]改成()
- 方法2 :yield关键字
2.1 生成式改写
- 通过for循环,依次计算并生成每一个元素
- 如果需要一个一个打印出来,可以通过next()函数获得生成式的下一个返回值
- 怎么让生成式进行计算? 1)next(g) 2)通过for循环遍历生成器对象
1)使用next(g)
g就是生成器对象,一边循环一边计算,在使用生成器时,在我们需要的时候,才会计算并生成
2)使用for循环遍历生成式对象
g = (x**2 for x in range(10))
for item in g:
print(item)
2.2 yield关键字
1)如果函数中包含yield关键字,调用函数的返回值是生成器
# yield(停靠,让步)
def welcome():
while True:
print('step 1')
# 函数中包含yield关键字,调用函数的返回值是生成器
yield 'ok'
print('step 2')
result = welcome()
print(result)
2)使用next函数执行调用
通过调用next函数才开始执行函数的内容,遇到yield就停止执行。当再次调用next方法时,从上次停止的地方执行,直到遇到yield关键字继续停止。
- yield是暂停本次
- 再次调用next方法时,会从上次停止的地方执行
3)用for循环实现
vim 04_Generator.py
- 编译
3. 函数中return和yield的区别是什么?
函数中遇到return就结束执行;
如果在函数中包含yield关键字,调用函数的返回值则是生成器
3.1 return
# 理解return关键字的工作原理
def welcome():
while True:
print('step 1')
# 函数中遇到return,函数执行结束,后面的代码不会继续执行
return 'ok'
print('step 2')
result = welcome()
print(result)
没有出现step 2
是因为遇见return
函数就会结束执行
3.2 yield
# yield(停靠,让步)
def welcome():
while True:
print('step 1')
# 函数中遇到return,函数执行结束,后面的代码不会继续执行
yield 'ok'
print('step 2')
result = welcome()
print(result)
结果显示为生成器