文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

一日一技:生成器里面的 return 有什么用?

2024-12-03 04:49

关注

本文转载自微信公众号「未闻Code」,作者kingname 。转载本文请联系未闻Code公众号。

在粉丝群里,有一位同学问了这样一个问题:

  1. def gen_data(num): 
  2.     if num > 10: 
  3.         for i in range(num): 
  4.             yield i 
  5.     else
  6.         return num 
  7.  
  8. generator = gen_data(5) 
  9. for num in generator: 
  10.     print(num) 

当传入的的参数小于等于10 的时候,为什么没有返回这个参数本身?

这道题,当我们传入的参数大于10的时候,能得到符合预期的结果,如下图所示:

但是,当我们传入数据5的时候,我们来看看运行效果:

可以看到,数字5并没有被打印出来,程序直接运行到了最后。

之所以会出现这种情况,是因为这个同学以为,当参数大于10的时候,gen_data(12)返回的是生成器,而当参数不大于10的时候,返回的是一个数字。显然这样的想法是不对的,否则,for num in 10这种语法早就报错了,数字是不能被迭代的。

正确的说法应该是,因为gen_data里面有yield,所以gen_data(参数)返回一个生成器。无论参数传入的是什么,返回的都是生成器。如下图所示:

为了说明为什么传入参数为5的时候,for 循环不执行,我们简化一下代码:

  1. def gen_data(): 
  2.     yield 1 
  3.     yield 2 
  4.     yield 3 
  5.     return 4 
  6.  
  7. generator = gen_data() 
  8. for num in generator: 
  9.     print(num) 

运行效果如下图所示:

可以看到,对于这样一个非常简单的生成器,在 for 循环里面也只是打印了数字123,并没有打印数字4。

关于生成器中的return,我们可以从 Python 官方文档PEP 255 — Simple Generators[1]中找到说明:

return 在生成器中,表示生成器运行完成了,可以结束了。然后生成器会抛出一个StopIteration的异常。而for循环能够检测到这个异常,于是结束循环。所以当我们传入的参数为5的时候,生成器直接运行到了 return,于是它直接就抛出StopIteration,于是 for 循环检测到这个异常就结束了。

在生成器里面的return只是一个结束标志,它不会把后面写的值返回给调用者。这跟函数里面的return语句是不一样的。

[1]PEP 255 — Simple Generators: https://www.python.org/dev/peps/pep-0255/#specification-return

 

来源: 未闻Code内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯