Python中如何根据传入参数数量递归实现二元操作泛化函数

完全可以实现，你要的逻辑本质是对任意数量输入的右嵌套二元运算折叠，在Python中通过收集不定长位置参数的方式就能搞定，下面是两种可直接落地的实现：

方案1：递归实现（逻辑最贴合需求） #

用*args接收任意数量的入参，通过递归逐层嵌套调用，和你描述的嵌套结构完全对应：

def foo(*args):
    # 入参校验：二元运算最少需要2个参数
    if len(args) < 2:
        raise ValueError("调用至少需要传入2个参数")
    # 递归终止条件：剩余2个参数时直接执行原生二元运算
    if len(args) == 2:
        return operation(args[0], args[1])
    # 递归匹配嵌套逻辑：第一个参数 与 剩余参数的嵌套计算结果 做运算
    return operation(args[0], foo(*args[1:]))

这个实现的计算结果完全匹配要求：

• 传入2个参数：直接返回operation(arg1, arg2)，和原有函数行为完全兼容
• 传入3个参数：返回operation(arg1, operation(arg2, arg3))
• 传入4个及以上参数：按从右到左的嵌套逻辑逐层计算，和你举例的结构完全一致

方案2：迭代实现（适配超多入参的场景） #

递归实现如果传入参数超过1000个，会触碰到Python默认的递归栈深度限制，此时可以用迭代的方式从右往左累积计算结果，性能更高也没有栈溢出风险：

def foo(*args):
    if len(args) < 2:
        raise ValueError("调用至少需要传入2个参数")
    # 从最右侧的两个参数开始初始化计算结果
    res = operation(args[-2], args[-1])
    # 从倒数第三个参数开始向左遍历，逐次嵌套计算
    for val in reversed(args[:-2]):
        res = operation(val, res)
    return res

注意事项 #

• 两种实现都是严格按照右嵌套的顺序计算的，如果你的operation是不满足结合律的运算（比如减法、除法），计算结果会和嵌套顺序强绑定，和你需求中的逻辑完全对齐
• 原有函数的调用兼容性是完全保障的：只传2个参数时，两种实现的返回结果和原函数没有任何区别，不会影响已有调用的逻辑

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