一个函数，它接受一个单一的参数或一组元组并将其视为多个参数。

Question

存在一种情况，我有一个获取函数，它返回一组项目的元组。我有一个设置函数，并希望它能接受单个项或者作为多个参数处理该元组的结果。但是，如果我在设置函数中使用 *args 作为参数以便接受可变数量的参数，那么它在两种情况下都不能按预期工作。

def get():
    return 0, 1, 2, 3

def set(*numbers):
    for i in range(len(numbers)):
        print(numbers[i])

single = 0
multiple = get()
set(single)
set(multiple)

set(single) 按预期工作并打印一个 0，而 set(multiple) 却未能按预期工作，它打印了一个 (0, 1, 2, 3) 元组，而不是遍历这些值并分别打印 0, 1, 2, 3。在用户不必手动解构 get 函数返回结果后再传递给 set 函数的情况下，最优雅的解决方案是什么？

John Millikin · Answer

正如评论中回答的那样：set(*get()) 可以很好地解决问题。

trashgod · Answer

由于 get 函数返回的是一个元组，你认为 set 函数应该接受多个变量，所以在参数 numbers 前添加了 *。但实际上 get 函数返回的是单一值，即元组 (0, 1, 2, 3) 视为一个整体。因此，set 函数只接收到一个参数，而这个参数本身就是一个元组，所以它会打印整个元组。只需移除参数 numbers 前的 * 符号，对于 set(get()) 的调用便会正常工作。但对于 set(0)，由于 0 不是一个元组，会导致错误。

鉴于你清楚地知道 set() 函数的参数可能是整数或元组，因此在 set() 函数内部提供一个条件语句更为合理。

def get():
    return (0, 1, 2, 3)

def set(numbers):
    if isinstance(numbers, int):  # 如果numbers是整数
        numbers = [numbers]  # 将其转换为包含单个元素的列表
    length = len(numbers)
    for i in range(length):
        print(numbers[i])

single = 0
multiple = get()
set(single)
set(multiple)