Python 生成器里面的 return 有什么用?

数据分析学习社
数据分析学习社
数据分析学习社
348
文章
0
评论
2021-07-2009:50:58 评论 96 1083字
摘要

Python 生成器里面的 return 有什么用

最近,在交流群里,有一位小伙伴问了这样一个问题:

def gen_data(num):    if num > 10:        for i in range(num):            yield i    else:        return numgenerator = gen_data(5)for num in generator:    print(num)

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

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

如下图所示:

Python 生成器里面的 return 有什么用?

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

Python 生成器里面的 return 有什么用?

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

之所以会出现这种情况,是因为这个同学以为,当参数大于 10 的时候,gen_data(12)返回的是生成器,而当参数不大于 10 的时候,返回的是一个数字

显然这样的想法是不对的,否则,for num in 10这种语法早就报错了,数字是不能被迭代的

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

如下图所示:

Python 生成器里面的 return 有什么用?

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

def gen_data():    yield 1    yield 2    yield 3    return 4generator = gen_data()for num in generator:    print(num)

运行效果如下图所示:

Python 生成器里面的 return 有什么用?

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

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

Python 生成器里面的 return 有什么用?

return 在生成器中,表示生成器运行完成了,可以结束了。然后生成器会抛出一个StopIteration的异常。而for循环能够检测到这个异常,于是结束循环

所以当我们传入的参数为 5 的时候,生成器直接运行到了 return,于是它直接就抛出StopIteration,于是 for 循环检测到这个异常就结束了

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

 

End.

作者:kingname

本文为转载分享,如果涉及作品、版权和其他问题,请联系我们第一时间删除(微信号:lovedata0520)

更多文章前往首页浏览http://www.itongji.cn/

  • 我的微信公众号
  • 微信扫一扫
  • weinxin
  • 我的微信公众号
  • 微信扫一扫
  • weinxin
匿名

发表评论

匿名网友 填写信息

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: