根据传递的参数数量更改返回值的数量[重复]

Question

是否有一种简短的方法可以对参数元组 *args 执行操作，如果将单个参数传递给函数，则返回单个参数，否则返回参数元组？ 如下例所示。

def transform(*args):
    if len(args) == 1:
        return args[0] + 1
    else:
        return tuple(arg + 1 for arg in args)

这样对于单个参数，您可以简单地调用转换而无需额外的解包：

x = transform(3)

以及多个参数：

a, b, c = transform(1, 2, 3)

另一种选择是说总是接受 Sized 并返回 Sized，

def transform_2(args):
    return tuple(arg + 1 for arg in args)

但这似乎使用户代码不那么直观：

x, = transform_2((3,))
a, b, c = transform_2((1, 2, 3))

或者这个解决方案更好吗？

Answer 1

总是采用可迭代的 API 更容易理解，并且不会给用户带来很大的负担。

def transform(args):
    return tuple(arg + 1 for arg in args)

x = transform((3,))  # or transform([3])
a, b, c = transform([1,2,3])

如果您尝试传递一个字符串，任何尝试做您想做的事情都会失败，因为您可能打算 foo("123") 返回单个值，而不是 foo("1") 的结果、foo("2") 和 foo("3")，但没有明确测试 str，您将无法区分字符串参数和另一个可迭代对象。