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

举报
未闻Code 发表于 2022/04/08 22:27:51 2022/04/08
【摘要】 生成器中return的作用

图片

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

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
【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。