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

问题描述 投票:2回答:1

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

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

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

python python-3.x design-patterns tuples
1个回答
4
投票

你不应该在长度为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
© www.soinside.com 2019 - 2024. All rights reserved.