在python中有条件地产生一个生成器[重复]

Question

假设我有这个代码：

def f(data, all_at_once):
    if all_at_once:
        return data
    else:
        yield from data

f([1,2,3], True)

f 总是返回一个生成器，不管all_at_once 的值如何。如何获得有条件返回生成器的函数？

Answer 1

这是一个开始。也许可以通过lambda来改进

def gen(data):
    yield from data

def f(data, all_at_once):
    if all_at_once:
        return data
    return gen(data)

f([1,2,3], True)