在python中获得收益后的回报值[重复]

Question

我想知道在这样的函数中执行完所有yield之后如何获取函数的返回值：

def gen_test():
    l = []
    for i in range(6):
        l.append(i)
        yield i
    # i want to know this value after all iteration of yield
    return l

Answer 1

简短的版本是不允许的。 在生成器中使用 return 语句传递值会导致 Python 3.3 版本之前的错误。对于这些版本的 Python，return 只能在没有表达式列表的情况下使用，等效于raise StopIteration。

对于更高版本的Python，可以通过异常的value-attribute来提取返回值。

您可以在此处找到更多信息：Return and yield in the same function

Answer 2

好的，这个怎么样。不要从您的生成器中return，产生列表值l 作为您最后一次返回控制权。这样你就可以避免人们建议的所有 Python 卷积，然后继续你的一天。

您始终可以测试生成器的返回类型以查看其是否为列表。这在 Python 中被认为是完全有效的。

def gen_test():
    l = []
    for i in range(6):
        l.append(i)
        yield i
    yield l

for y in gen_test():
    if isinstance( y, type([])):
        print("List is ", y)
    else:
        print (y)
    

Answer 3

您不能轻易在生成器函数（包含yield 的函数）中使用return 来返回表达式，因为调用该函数会返回生成器。生成器中的return 与提高StopIteration 相同，如果您返回一个值，则会将其填充到 ecxeption 中。由于 Python 中的大多数迭代方式都不允许您访问 StopIteration，因此很难获得此值，除非您避开 for 和大多数其他迭代方式（例如 list()）。

您可以使用所需的返回值引发异常：

class ReturnValue(Exception):
    pass

def gen_test():
    l = []
    for i in range(6):
        l.append(i)
        yield i
    raise ReturnValue(l)

try:
    for x in gen_test():
        print(x)
except ReturnValue as e:
    print(e.args[0])  # or
    returnvalue = e.args[0]
    # note: variable e goes away after exception handler

不过，这似乎是一个 hack。一个类可能会更好：

class GenTest:
    value = None
    def __iter__(self):
        l = []
        for i in range(6):
            l.append(i)
            yield i
        self.value = l

for x in (it := GenTest()):
    print(x)

print(it.value)

Answer 4

迭代器函数的返回值用作它在终止时引发的StopIteration 的参数：

>>> it = gen_test()
>>> next(it)
0
>>> next(it)
1
>>> next(it)
2
>>> next(it)
3
>>> next(it)
4
>>> next(it)
5
>>> next(it)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration: [0, 1, 2, 3, 4, 5]

只有第一个StopIteration 获得指定的值。进一步尝试迭代空生成器将引发空异常：

>>> next(it)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

您可以通过避免for 循环并自己捕获异常来从迭代器的正常运行中提取返回值：

it = gen_test()
while True:
    try:
        x = next(it)
        # Do the stuff you would normally do in a for loop here
    except StopIteration as e:
        ret = e.value
        break
print(ret)

鉴于您必须在不使用 for 循环和无法返回累积的数据之间做出决定，所有这些都是一种尴尬的方法。假设您不同意尝试在外部累积数据（例如，只是做list(gen_test())），您可以使用类公开您的生成器状态：

class gen_test:
    def __init__(self):
        self.l = []
        self.it = iter(range(6))

    def __iter__(self):
        return self

    def __next__(self):
        i = next(self.it)
        self.l.append(i)
        return i

现在你可以做类似的事情

>>> it = gen_test()
>>> for x in it:
...     print(x)
0
1
2
3
4
5
>>> print(it.l)
[0, 1, 2, 3, 4, 5]

Answer 5

yield 在你的函数中的存在使它变成了不同于常规函数的东西——python 称之为generator。您不能调用生成器并期望它像函数一样执行。除非在退化或简单化的情况下。 要在生成器生命周期结束时检索问题中的状态信息（如整个列表 l），您需要将其存储在变量中或每次在 yield 语句中返回。

要在每次调用生成器时返回它：

def gen_test():
    l = []
    for i in range(6):
        l.append(i)
        yield i, l
    
for y,l in gen_test():
        print (y,l)
    

否则，您可以将生成器声明为类的一部分并将中间结果存储在类变量中。然后稍后使用不同的调用检索该变量值。