Question

我正在运行python 2.7.3。我有一个我调用的函数列表。某些函数返回多个输出。我想将所有功能的所有输出保存在列表中。我怎么能这样做？

def f1():
    return [1,2], [3,4]
def f2():
    return [5,6]
my_outputs = []
my_funcs = [f1, f2]
for func in my_funcs:
    output_list* = func() # does this work?
    # a,b,c... = func()
    my_outputs.extend(output_list)

print my_outputs
[[1,2], [3,4], [5,6]]

Answer 1

您可以使用isinstance检查返回类型，但它不是非常pythonic。

 my_outputs = []
 my_funcs = [f1, f2, f3]
 for func in my_funcs:
    result = func()
    if isinstance(result, tuple):
        my_outputs.extend(result)
    else:
        my_outputs.append(result)

Answer 2

sum((list(f()) for f in [f1, f2]), [])

如何读取可变数量的输出

2 个答案: