python 中函数链的探索

在 codewars.com 上，我们遇到的一个任务是创建一个可以连续相加数字的函数 add。然而，我们困惑于如何实现这种连续调用的功能。

函数链的本质

本质上，函数链允许函数像这样连续调用：f(x)(y)(z)...。这类似于数学中对函数的理解，其中 f(x) 将每个 x 映射到函数 g_x，然后返回 g_x(y)。

立即学习“Python免费学习笔记（深入）”；

在 python 中实现函数链

python 查询技能

查询客流数据，输出JSON格式，可直接导入Bitable等可视化工具

下载

在 python 中，有两种实现函数链的方法：

1. 子分类 int 并定义 call

我们可以创建自定义的 int 子类并定义其 call 方法，使其返回一个具有更新值的自身新实例：

class customint(int):
    def __call__(self, v):
        return customint(self + v)

add = customint

print(add(1))  # 1
print(add(1)(2))  # 3
print(add(1)(2)(3)(44))  # 50

2. 定义一个闭包

另一种方法是创建一个嵌套函数，需要一个额外的空参数调用才能返回结果：

def add(v):
    def _inner_adder(val=None):
        if val is None:
            return _inner_adder.v
        _inner_adder.v += val
        return _inner_adder
    _inner_adder.v = v  # save value
    return _inner_adder

print(add(1)(2)())  # 3
print(add(1)(2)(3)())  # 6

这两种方法都允许我们创建可以连续调用的函数，从而扩展了 python 中函数使用的可能性。