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

Question

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

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

我正在使用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中产生每个项目,所以我可以返回在另一个函数中创建的生成器,但是有更好的方法吗？

Answer 1

如何使用生成器表达式？

>>> 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(start, stop, step))代替@DSM评论的生成器表达式.请参阅内置功能 -iter.

Answer 2

falsetru为您提供了一种返回生成器或列表的函数的方法.我正在回答你的另一个问题,即为什么你不能在同一个函数中获得回报和收益.

当你调用生成器函数时,它实际上并没有立即执行任何操作(请参阅此问题).它不执行函数体,但等待直到你开始迭代它(或调用next它).因此,当你调用函数时,Python必须知道函数是否是生成器函数,以了解是否运行函数体.

然后让它返回一些值是没有意义的,因为如果它是一个生成器函数它返回的是生成器(即,你迭代的东西).如果你可以在生成器函数内部返回,则在调用函数时将无法访问它,因此函数必须"自发地"在稍后的某个时间点返回一个值(当return生成器被消耗时达到语句时),或者与当时产生价值相同,或者是奇怪和令人困惑的东西.

我同意,假设有一个函数有时会返回一个生成器,有时候返回一个列表,这可能不是一个好主意.list如果你想要一个列表,只需调用生成器.