yield是Python中和生成器密切相关的一个关键字
那么什么是生成器呢:
- 生成器是一个不断产生值的函数
- 包含yield语句的函数是一个生成器
- 生成器每次产生一个值(yeild语句),函数就会被冻结,直到被唤醒后再产生一个值。
生成器写法:
def gen(n):
for i in range(n):
yield i**2
for i in gen(1000000):
print(i)
普通写法:
def square(n):
ls = [i**2 for i in range(n)]
return ls
for i in square(1000000):
print(i)
使用生成器的优势:
- 节省存储空间:生成器在每次调用的时候只产生一个值,所以只需要一个元素的存储空间;而普通写法则是产生一个列表,需要占用大量的存储空间
- 响应更加迅速
- 使用更灵活