Python中的柯里化：如何实现支持任意输入长度的加减操作函数

嗨，这个问题挺典型的！你写的原函数是个嵌套闭包，但只能支持两次调用（初始值加一次减法）。要实现任意次数的链式加减操作，我们可以用闭包保存状态，或者写个简单的类来管理当前值——两种方法都不用碰functools模块。

方案1：兼容原逻辑的连续减法（支持任意次数链式调用） #

如果你的核心需求是延续原函数的减法逻辑，只是想支持多次连续减操作，我们可以用闭包来保存当前的计算结果，同时让返回的函数可以被直接转换成数值：

def addsub(initial):
    # 用列表保存当前值（整数是不可变类型，闭包无法直接修改）
    current = [initial]
    
    def subtract(num):
        current[0] -= num
        # 返回自身，实现链式调用
        return subtract
    
    # 让函数可以被转换为整数
    subtract.__int__ = lambda self: current[0]
    # 直接打印时显示当前数值
    subtract.__repr__ = lambda self: str(current[0])
    
    return subtract

测试用法：

# 原调用方式，返回5
print(addsub(9)(4))  # 输出: 5

# 三次减法调用：9-3-4-5 = -3
print(addsub(9)(3)(4)(5))  # 输出: -3

方案2：支持混合加减的灵活版本 #

如果你需要同时支持加法和减法操作，用类来实现会更直观，还能兼容原有的调用方式：

class AddSub:
    def __init__(self, initial_value):
        self.current = initial_value
    
    def __call__(self, num):
        # 兼容原函数的减法逻辑，直接调用时执行减法
        self.current -= num
        return self
    
    def add(self, num):
        # 明确执行加法
        self.current += num
        return self
    
    def subtract(self, num):
        # 明确执行减法
        self.current -= num
        return self
    
    # 支持转换为数值类型
    def __int__(self):
        return self.current
    
    # 打印时直接显示结果
    def __repr__(self):
        return str(self.current)

测试用法：

# 原调用方式，返回5
print(AddSub(9)(4))  # 输出: 5

# 混合加减操作：9 +3 -4 +5 = 13
print(AddSub(9).add(3).subtract(4).add(5))  # 输出: 13

# 链式调用混合直接调用和方法：9-3+4-(-5)=15
print(AddSub(9)(3).add(4)(-5))  # 输出:15

原理说明 #

• 闭包方案：通过列表保存当前值（避开不可变类型的修改限制），每次调用更新值后返回自身，同时自定义特殊方法让函数可以被当作数值使用。
• 类方案：用实例属性保存当前值，通过__call__方法兼容原调用逻辑，同时提供明确的add/subtract方法实现灵活操作，链式调用的核心是每个方法都返回self。

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