python3中的可选yield或return。如何？

Question

我想要一个可以选择返回或产生结果的函数。 这是一个例子。

def f(option=True):
    ...
    for...:
        if option:
            yield result
        else:
            results.append(result)

    if not option:
        return results

当然，这是行不通的，我用 python3 试过了，不管我设置什么选项值，我总是得到一个生成器。 据我了解，python 检查函数的主体，如果存在yield，那么结果将是一个生成器。 有什么办法可以解决这个问题并制作一个可以随意返回或屈服的函数？

Answer 1

基于class 的方法

class FunctionAndGenerator:
    def __init__(self):
        self.counter = 0
        
    def __iter__(self):
        return self
        

# You need a variable to indicate if dunder next should return the string or raise StopIteration. 
# Raising StopIteration will stop the loop from iterating more.
# You'll have to teach next to raise StopIteration at some point

    def __next__(self):
        self.counter += 1 
        if self.counter > 1 :
            raise StopIteration 
        return f"I'm a generator and I've generated {self.counter} times"        

    def __call__(self):
        return "I'm a function"
        
x = FunctionAndGenerator()
print(x())
    
for i in x:
    print(i)

I'm a function
I'm a generator and I've generated 1 times

[Program finished] 

Answer 2

你不能。 任何使用yield 都会使函数成为生成器。

您可以使用一个使用list() 将生成器生成的所有值存储在列表对象中并返回的函数来包装您的函数：

def f_wrapper(option=True):
    gen = f()
    if option:
        return gen    # return the generator unchanged
    return list(gen)  # return all values of the generator as a list

但是，一般来说，这是糟糕的设计。不要让你的函数像这样改变行为；坚持一种返回类型（生成器或一个对象），不要让它在两者之间切换。

考虑将其拆分为两个函数：

def f():
    yield result

def f_as_list():
    return list(f())

如果您需要生成器，请使用f()，如果您想要一个列表，请使用f_as_list()。

由于list()（和next() 仅访问生成器的一个值）是内置函数，因此您很少需要使用包装器。直接调用这些函数即可：

# access elements one by one
gen = f()
one_value = next(gen)

# convert the generator to a list
all_values = list(f())

Answer 3

这个怎么样？

def make_f_or_generator(option):
    def f():
        return "I am a function."
    def g():
        yield "I am a generator."
    if option:
        return f
    else:
        return g

这至少让您可以选择创建函数或生成器。