Python 生成器里面的 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语句是不一样的

推荐阅读:知识星球: 菜鸟学投资，星球来了！入门: 最全的零基础学Python的问题  | 零基础学了8个月的Python  | 实战项目 |学Python就是这条捷径干货:爬取豆瓣短评，电影《后来的我们》 | 38年NBA最佳球员分析 |   从万众期待到口碑扑街！唐探3令人失望  | 笑看新倚天屠龙记 | 灯谜答题王 ｜用Python做个海量小姐姐素描图 ｜碟中谍这么火，我用机器学习做个迷你推荐系统电影趣味:弹球游戏  | 九宫格  | 漂亮的花 | 两百行Python《天天酷跑》游戏!AI: 会做诗的机器人 | 给图片上色 | 预测收入 | 碟中谍这么火，我用机器学习做个迷你推荐系统电影小工具: Pdf转Word，轻松搞定表格和水印！ | 一键把html网页保存为pdf！|  再见PDF提取收费！ | 用90行代码打造最强PDF转换器，word、PPT、excel、markdown、html一键转换 | 制作一款钉钉低价机票提示器！ ｜60行代码做了一个语音壁纸切换器天天看小姐姐！｜年度爆款文案1).卧槽！Pdf转Word用Python轻松搞定！2).学Python真香！我用100行代码做了个网站，帮人PS旅行图片，赚个鸡腿吃3).首播过亿，火爆全网，我分析了《乘风破浪的姐姐》，发现了这些秘密 4).80行代码！用Python做一个哆来A梦分身 5).你必须掌握的20个python代码，短小精悍，用处无穷 6).30个Python奇淫技巧集 7).我总结的80页《菜鸟学Python精选干货.pdf》,都是干货 8).再见Python！我要学Go了！2500字深度分析！9).发现一个舔狗福利！这个Python爬虫神器太爽了，自动下载妹子图片点阅读原文,领AI全套资料！