从长度为1的元组解包返回值

Posted 2021-05-06

假设我有一个虚函数，它可以返回确切的参数：

def dummy_single(arg1):
    """Repeat the same argument."""
    return arg1

我可以通过以下方式使用此功能：

same_obj1 = dummy_single(obj1)

现在，如果我想允许此函数接受多个参数，并返回相同的参数，我可以打包并解压缩参数：

def dummy_multiple(*args):
    """Repeat all given arguments."""
    return args

请注意，此功能恰好是身份，但这不是问题的核心。你可以想象任何函数返回一个可变长度的元组。

使用多个参数调用它时效果很好：

same_obj1, same_obj2 = dummy_multiple(obj1, obj2)

但是，这不会按预期工作：

same_tuple = dummy_multiple(obj1)

same_obj1包含一个（打包的）元组，而不是前两个示例中的对象。我考虑过以下方式修复此行为，但它看起来并不干净，因为返回的类型因参数而异。

def dummy_multiple_fixed(*args):
    """Repeat all given arguments."""
    if len(args) == 1:
        return args[0]
    return args

这不是一个关键问题，很容易修复。但我找不到一个干净而优雅的方法来解决这个问题。

答案

你不应该在长度为1的元组中做出特殊情况。函数应该有一个明确定义的返回类型。

虽然，知道你可以解压缩长度为1的元组：

def dummy_multiple(*args):
    """Repeat all given arguments."""
    return args

same_obj1, = dummy_multiple(1)
#        ^ notice the comma here

print(same_obj1) # 1