函数式编程 - partial 函数

优质

小牛编辑

131浏览

2023-12-01

Python 提供了一个 functools 的模块，该模块为高阶函数提供支持，partial 就是其中的一个函数，该函数的形式如下：

functools.partial(func[,*args][, **kwargs])

这里先举个例子，看看它是怎么用的。

假设有如下函数：

def multiply(x, y):
    return x * y

现在，我们想返回某个数的双倍，即：

>>> multiply(3, y=2)
6
>>> multiply(4, y=2)
8
>>> multiply(5, y=2)
10

上面的调用有点繁琐，每次都要传入 y=2，我们想到可以定义一个新的函数，把 y=2 作为默认值，即：

def double(x, y=2):
    return multiply(x, y)

现在，我们可以这样调用了：

>>> double(3)
6
>>> double(4)
8
>>> double(5)
10

事实上，我们可以不用自己定义 double，利用 partial，我们可以这样：

from functools import partial
double = partial(multiply, y=2)

partial 接收函数 multiply 作为参数，固定 multiply 的参数 y=2，并返回一个新的函数给 double，这跟我们自己定义 double 函数的效果是一样的。

所以，简单而言，partial 函数的功能就是：把一个函数的某些参数给固定住，返回一个新的函数。

需要注意的是，我们上面是固定了 multiply 的关键字参数 y=2，如果直接使用：

double = partial(multiply, 2)

则 2 是赋给了 multiply 最左边的参数 x，不信？我们可以验证一下：

from functools import partial
def subtraction(x, y):
    return x - y
f = partial(subtraction, 4)  # 4 赋给了 x
>>> f(10)   # 4 - 10
-6

小结