Python 列表为 *args？

Question

我有两个 Python 函数，它们在函数定义中都采用变量参数。举一个简单的例子：

def func1(*args):
    for arg in args:
        print arg

def func2(*args):
    return [2 * arg for arg in args]

我想组合它们 - 就像 func1(func2(3, 4, 5)) 一样 - 但我不想要 args 在 func1 中为 ([6, 7, 8],)，我希望它为 (6, 7, 8)，就好像它被称为 func1(6, 7, 8) 而不是 func1([6, 7, 8]) 一样。

通常，我只使用 func1(*func2(3, 4, 5)) 或让 func1 检查是否 args[0]是一个清单。不幸的是，我无法在这个特定实例中使用第一个解决方案，并且应用第二个解决方案需要在许多地方进行这样的检查（有很多函数充当 func1 的角色）。

有人知道如何做到这一点吗？我想可以进行某种内省，但我可能是错的。

Answer 1

您可以考虑编写函数装饰器来检查第一个参数是否是列表。将装饰器应用于现有函数比修改函数要简单一些。

Answer 2

您可以使用 Yaroslav 发布的装饰器。

最小的例子：

def unpack_args(func):
    def deco_func(*args):
        if isinstance(args, tuple):
            args = args[0]

        return func(*args)

    return deco_func


def func1(*args):
    return args

def func2(*args):
    return args

@unpack_args
def func3(*args):
    return args

print func1(1,2,3)    # > (1,2,3)
print func2(1,2,3)    # > (1,2,3)
print func1(*func2(1,2,3))    # > (1,2,3)
print func1(func2(1,2,3))    # > ( (1,2,3), )
print func3(func2(1,2,3))   # > (1,2,3)

Answer 3

通常情况下，我只会使用 func1(*func2(3, 4, 5)) 或让 func1 检查 args[0] 是否为 列表。不幸的是，我无法在这个特定实例中使用第一个解决方案，并且应用第二个解决方案需要在许多地方进行这样的检查（有很多函数充当 func1 的角色）。< /p>


为什么不能使用第一个解决方案？

>>> def func1(*args):
    for arg in args:
        print arg

>>> def func2(*args):
    return [2 * arg for arg in args]

>>> func2(3, 4, 5)
[6, 8, 10]
>>> func1(1,2,3)
1
2
3
>>> func1(*func2(3, 4, 5))
6
8
10
>>> 

如果不可能，您仍然可以使用 func1(*tuple(func2(3, 4, 5))) （但您不需要）。