当前栏目
一日一技:生成器里面的 return 有什么用?
本文转载自微信公众号「未闻Code」,作者kingname 。转载本文请联系未闻Code公众号。
在粉丝群里,有一位同学问了这样一个问题:
- def gen_data(num):
- if num > 10:
- for i in range(num):
- yield i
- else:
- return num
- generator = gen_data(5)
- for num in generator:
- print(num)
当传入的的参数小于等于10 的时候,为什么没有返回这个参数本身?
这道题,当我们传入的参数大于10的时候,能得到符合预期的结果,如下图所示:
但是,当我们传入数据5的时候,我们来看看运行效果:
可以看到,数字5并没有被打印出来,程序直接运行到了最后。
之所以会出现这种情况,是因为这个同学以为,当参数大于10的时候,gen_data(12)返回的是生成器,而当参数不大于10的时候,返回的是一个数字。显然这样的想法是不对的,否则,for num in 10这种语法早就报错了,数字是不能被迭代的。
正确的说法应该是,因为gen_data里面有yield,所以gen_data(参数)返回一个生成器。无论参数传入的是什么,返回的都是生成器。如下图所示:
为了说明为什么传入参数为5的时候,for 循环不执行,我们简化一下代码:
- def gen_data():
- yield 1
- yield 2
- yield 3
- return 4
- generator = gen_data()
- for num in generator:
- 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
相关文章
- 鲜为人知但很有用的 HTML 属性
- 翻转再翻转!有意思的水平横向溢出滚动
- 自定义计数器小技巧!CSS 实现长按点赞累加动画
- 过五关!React高频面试题指南
- 软件开发中的十个认知偏差
- 不需要 JS!仅用 CSS 也能达到监听页面滚动的效果!
- 一文读懂TypeScript类型兼容性
- Vue 的响应式原则与双向数据绑定
- 快速掌握 TypeScript 新语法:Infer Extends
- JWT教你如何证明你是我的人!
- 一篇带给你 V8 GC 的实现
- 面试官:请使用JS完成一个LRU缓存?
- 通过可视化来学习JavaScript事件循环
- 新的跨域策略:使用 COOP、COEP 为浏览器创建更安全的环境
- 为什么有人说 vite 快,有人却说 vite 慢?
- 种草 Vue3 中几个好玩的插件和配置
- 超全面的前端工程化配置指南
- Vue 状态管理未来样子
- Volatile关键字能保证原子性么?
- 面试突击:SpringBoot 有几种读取配置文件的方法?