什么时候使用return(产量)?

问题描述 投票:22回答:1

很长一段时间,我不知道你不能把return放在收益率声明面前。但实际上你可以:

def gen():
    return (yield 42)

这类似于

def gen():
    yield 42
    return

我能想到的唯一用法是将发送的值附加到StopIterationpep-0380

生成器中的return expr导致从生成器退出时引发StopIteration(expr)。

def gen():
    return (yield 42)

g = gen()
print(next(g))  # 42
try:
    g.send('AAAA')
except StopIteration as e:
    print(e.value)  # 'AAAA'

但这也可以使用额外的变量来完成,这更明确:

def gen():
    a = yield 42
    return a

g = gen()
print(next(g))
try:
    g.send('AAAA')
except StopIteration as e:
    print(e.value)  # 'AAAA'

因此,似乎return (yield xxx)仅仅是一种语法糖。我错过了什么吗?

python generator
1个回答
7
投票

在生成器内部,表达式(yield 42)将产生值42,但如果使用next(generator),则返回一个值为None,如果使用generator.send(value),则返回给定值。

正如你所说,你可以使用一个中间值来获得相同的行为,不是因为这是语法糖,而是因为yield表达式实际上是返回你发送它的值。

你也可以这样做

def my_generator():
    return (yield (yield 42) + 10)

如果我们调用它,使用调用序列:

g = my_generator()
print(next(g))
try:
    print('first response:', g.send(1))
    print('Second response:', g.send(22))
    print('third response:', g.send(3))
except StopIteration as e:
    print('stopped at', e.value)

首先我们得到42的输出,并且生成器基本上暂停在你可以描述的状态:return (yield <Input will go here> + 10),如果我们然后调用g.send(1)我们得到输出11.并且生成器现在处于状态:return <Input will go here>,然后发送g.send(22)将抛出一个StopIteration(22),因为在发电机中处理返回的方式。所以你永远不会因为异常而进入第三次发送。

我希望这个例子让yield如何在生成器中工作以及为什么语法return (yield something)没有什么特别或异国情调,并且正如你期望的那样工作。

至于字面问题,你什么时候会这样做?好吧,当你想要产生某些东西,然后返回一个StopIteration回显发送给发生器的用户的输入。因为这实际上是代码所说的内容。我希望这种行为很少被人想要。

© www.soinside.com 2019 - 2024. All rights reserved.