Python函数返回列表还是作为生成器工作？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了Python函数返回列表还是作为生成器工作？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1362字，纯文字阅读大概需要2分钟。

内容图文

我正在尝试创建一个返回对象或作为生成器的函数.

这是一个坏主意,因为作为一种最佳实践,您希望函数可靠地返回相同类型的值,但为了科学的利益……

我正在使用Python 2,因此range返回一个列表,xrange是一个可迭代的(有趣的是它也提供了__len__).

def xr(start, stop=None, step=1, gen=True):
    if stop is None:
        start, stop = 0, start
    if gen == True:
        for i in xrange(start, stop, step):
            yield i
    else:
        return range(start, stop, step)

我收到此错误：

  File "<stdin>", line 8
SyntaxError: 'return' with argument inside generator

问题：

为什么(除了显而易见的“你不能同时在函数中产生和返回”,假设是正确的)它是否这样做？看看我的代码,不清楚为什么这样做会很糟糕.

我怎么试图绕过这个？我知道我可以返回xrange而不是从xrange中产生每个项目,所以我可以返回在另一个函数中创建的生成器,但是有更好的方法吗？

解决方法:

使用generator expression怎么样？

>>> def xr(start, stop=None, step=1, gen=True):
...     if stop is None:
...         start, stop = 0, start
...     if gen == True:
...         return (i for i in xrange(start, stop, step)) # <----
...     else:
...         return range(start, stop, step)
...
>>> xr(2, gen=False)
[0, 1]
>>> xr(2, gen=True)
<generator object <genexpr> at 0x0000000002C1C828>
>>> list(xr(2, gen=True))
[0, 1]

顺便说一下,我宁愿只定义一个生成器函数.然后使用list(xr(..))如果我需要一个列表.

更新另外,您可以使用iter(xrange(开始,停止,步))而不是@DSM评论的生成器表达式.见Built-in functions — iter.

内容总结

以上是互联网集市为您收集整理的Python函数返回列表还是作为生成器工作？全部内容，希望文章能够帮你解决Python函数返回列表还是作为生成器工作？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端