一日一技：生成器里面的 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

免费源码下载

本文地址：http://www.bzuk.cn/html/02f32799670.html