PHP前端开发

Python 中如何实现函数链式调用?

百变鹏仔 5天前 #Python
文章标签 链式

python 中函数链的探索

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

函数链的本质

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

立即学习“Python免费学习笔记(深入)”;

在 python 中实现函数链

在 python 中,有两种实现函数链的方法:

1. 子分类 int 并定义 call

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

class customint(int):    def __call__(self, v):        return customint(self + v)add = customintprint(add(1))  # 1print(add(1)(2))  # 3print(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_adderprint(add(1)(2)())  # 3print(add(1)(2)(3)())  # 6

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