函数functools.partial()允许我们给一个或多个参数指定固定的值,以此减少需要提供给之后调用的参数数量。
举个例子:
def spam(a, b, c, d):
print(a, b, c, d)
s1 = functools.partial(spam, 1) # a = 1
s1(2, 3, 4) # 1 2 3 4
s2 = functools.partial(spam, d=42)
s2(1, 2, 3) # 1 2 3 42
原理:
1、functools.partial()对特定的参数赋了固定值并回了返一个全新的可调用对象;
2、这个新的可调用对象将传递给partial()的固定参数结合起来,统一将所有的参数传递给原始的函数;
下面以一个例子说明:有一列以元组(x, y)来表示点的坐标
points = [(1, 2), (3, 4), (5, 6), (7, 8)]
import math
def distance(p1, p2):
x1, y1 = p1
x2, y2 = p2
return math.hypot(x2-x1, y2-y1)
解法1:现在想根据它们之间的距离来对它们进行排序:列表的sort():
pt = (4, 3)
points.sort(key=functools.partial(distance, pt)) # partial()指定默认值pt
print(points)
解法2:有时候可以通过lambda表达式来替代
points.sort(key=lambda p: distance(pt, p))