Python多参数函数适配：支持可变参数与数组输入的实现方案

我来帮你搞定这个问题！你遇到的错误是因为当传入单个数组时，*args会把这个数组打包成一个只包含一个元素的元组，导致后续sum操作时试图把整数和列表相加，自然就报错了。

问题原因分析 #

当你调用arrSum([1,2,3])时，args的值是([1,2,3],)（一个元组，里面只有一个列表元素）。你原来的代码把args转成列表后extend，最终listOfArgs变成了[[1,2,3]]。sum函数会默认从0开始累加，尝试执行0 + [1,2,3]，这就触发了TypeError——整数和列表不能直接相加。

解决方案 #

我们可以通过判断输入的情况，区分“传入多个单个参数”和“传入一个数组”两种场景：

def arrSum(*args):
    # 情况1：传入单个列表/元组（排除字符串，避免把字符串误判为数组）
    if len(args) == 1 and isinstance(args[0], (list, tuple)):
        return sum(args[0])
    # 情况2：传入多个单个参数，直接sum可变参数
    return sum(args)

测试验证 #

• 调用arrSum(1,2,3)：args是(1,2,3)，直接sum(args)返回6，符合预期。
• 调用arrSum([1,2,3])：args是([1,2,3],)，满足len(args)==1且元素是列表，所以sum(args[0])返回6，符合预期。

如果你想支持更灵活的输入（比如同时传入多个数组），可以进一步扩展逻辑，但上面的代码已经完美覆盖你提到的两种核心场景啦。

内容的提问来源于stack exchange，提问作者Rohit Kumar