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

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

python 中函数链的探索

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

函数链的本质

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

python 中实现函数链

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 中函数使用的可能性。

以上就是Python 中如何实现函数链式调用?的详细内容,更多请关注其它相关文章!