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

如何使用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(..))。

UPDATE 或者，您可以使用iter(xrange(start, stop, step))代替@DSM评论的生成器表达式。请参阅Built-in functions -- iter。

Answer 2

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

当您调用生成器函数时，它实际上并没有立即执行任何操作（请参阅this question）。它不会执行函数体，但会等到你开始迭代它（或在它上面调用next）。因此，当你调用函数时，Python必须知道函数是否是生成器函数，以了解是否运行函数体。

然后让它返回一些值是没有意义的，因为如果它是一个生成器函数它返回的是生成器（即你迭代的东西）。如果你可以在生成器函数内部返回，那么当你调用函数时它就不会被触及，所以函数必须自发地＆＃34;稍后返回一个值（当生成器被消耗时达到return语句时），这或者与当时产生值相同，或者是奇怪和令人困惑的东西。

我同意假设一个函数有时候会返回一个生成器，有时甚至是一个列表，这可能不是一个好主意。如果您想要一个列表，只需在生成器上调用list。

Answer 3

你不能同时拥有这两者，因为解释器 - 遇到关键字 yield 时 - 将该函数视为生成器函数。显然你不能在生成器函数中有一个return语句。附加点 - 您不直接使用生成器函数，您可以使用它们来创建（我很想说实例化）生成器。简单的例子

def infinite():
    cnt = 1
    while True:
        yield cnt
        cnt += 1

infinite_gen = infinite()

无限是生成器函数， infinite_gen 是生成器

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

问题：

3 个答案: