Python 3.8：具有if / else条件和return / yield的定义始终返回生成器

如何解决Python 3.8：具有if / else条件和return / yield的定义始终返回生成器

[ Python 3.8 | Spyder ]

我试图引入一个简单的定义，该定义将按顺序return [1,2,3]或yield这些数字。

最低工作代码如下：

def example(condition):
    if condition:
        yield 1
        yield 2
        yield 3
    else:
        return [1,3]

如果尝试使用def，则它将始终返回生成器。即使return在if / else对中首先出现：

def example(condition):
    if not condition:
        return [1,3]

    else:
        yield 1
        yield 2
        yield 3

即使condition = False，似乎python也不忽略收益率的存在。 这是意想不到的行为。

解决方法

您可以以生成器表达式的形式返回生成器：

>>> def example(condition):
...     if condition:
...         return (i for i in (1,2,3))
...     else:
...         return [1,3]

>>> example(1)
<generator object example.<locals>.<genexpr> at 0x000000000257BBA0>

>>> example(0)
[1,3]

或者您可以单独定义生成器：

>>> def g():
...     yield 1
...     yield 2
...     yield 3
...
>>> def example(condition):
...     if condition:
...         return g()
...     else:
...         return [1,3]

>>> example(1)
<generator object g at 0x000000000257BBA0>

>>> example(0)
[1,3]

,

从语法上讲，套件中带有def语句的yield语句定义了一个生成器。是否在运行时实际到达yield是无关紧要的，实际上是不确定的。

def another_example():  # this function has a well-defined type
    if random.random() < 0.5:
       yield 'What is the type of this function?'
    return 'Of course it is a generator function!'

---

为了使函数评估为生成器值或非生成器值，请对生成器或generator-expression使用单独的定义。

def example_generator():
    yield 1
    yield 2
    yield 3

def example(condition):
    if not condition:
        return [1,3]
    else:
        return example_generator()

如果两个路径最终都应提供相同的值，则在功能之外进行选择在语义上是相同的。这使得仅提供生成器功能就足够了。结果可以在外部转换为另一种类型。

def example():
    yield 1
    yield 2
    yield 3

print("I prefer lists,such as:",list(example()))