Python中的生成器的创建方法 (生成式改写 + yield关键字)

本文介绍了Python中的生成器,包括其在处理大量数据时节省内存的优势,以及两种创建方法:生成式改写(将列表生成式改为括号)和使用yield关键字。详细阐述了yield关键字的作用,它使函数变成生成器,每次调用next()或for循环时,函数从上次yield处继续执行,而非重新开始。此外,对比了return和yield的区别,return会结束函数执行,而yield则生成生成器。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在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)

结果显示为生成器
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值