大家好，我是菜鸟哥！
最近，在交流群里，有一位小伙伴问了这样一个问题：
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语句是不一样的