偏导数非常有用。
例如,在“管道式”函数调用序列中(其中一个函数的返回值是传递给下一个函数的参数)。
有时,这样的管道中的函数需要一个参数,但它的上游函数返回两个值。
在这个场景中,使用functools。Partial可能允许您保持这个函数管道的完整性。
这里有一个具体的孤立的例子:假设你想要根据每个数据点到某个目标的距离对一些数据进行排序:
# create some data
import random as RND
fnx = lambda: RND.randint(0, 10)
data = [ (fnx(), fnx()) for c in range(10) ]
target = (2, 4)
import math
def euclid_dist(v1, v2):
x1, y1 = v1
x2, y2 = v2
return math.sqrt((x2 - x1)**2 + (y2 - y1)**2)
要根据与目标的距离对这些数据进行排序,你需要做的当然是:
data.sort(key=euclid_dist)
但你不能——sort方法的key形参只接受接受单个参数的函数。
所以把euclid_dist重写成一个带单参数的函数:
from functools import partial
p_euclid_dist = partial(euclid_dist, target)
P_euclid_dist现在接受一个参数,
>>> p_euclid_dist((3, 3))
1.4142135623730951
所以现在你可以通过传入sort方法的key参数的partial函数来排序你的数据:
data.sort(key=p_euclid_dist)
# verify that it works:
for p in data:
print(round(p_euclid_dist(p), 3))
1.0
2.236
2.236
3.606
4.243
5.0
5.831
6.325
7.071
8.602
或者,例如,函数的一个参数在外部循环中发生了变化,但在内部循环的迭代中是固定的。通过使用partial,您不必在内部循环迭代期间传递额外的参数,因为修改后的(partial)函数不需要它。
>>> from functools import partial
>>> def fnx(a, b, c):
return a + b + c
>>> fnx(3, 4, 5)
12
创建一个局部函数(使用关键字arg)
>>> pfnx = partial(fnx, a=12)
>>> pfnx(b=4, c=5)
21
您还可以使用位置参数创建偏函数
>>> pfnx = partial(fnx, 12)
>>> pfnx(4, 5)
21
但这将抛出(例如,使用关键字参数创建partial,然后使用位置参数调用)
>>> pfnx = partial(fnx, a=12)
>>> pfnx(4, 5)
Traceback (most recent call last):
File "<pyshell#80>", line 1, in <module>
pfnx(4, 5)
TypeError: fnx() got multiple values for keyword argument 'a'
另一个用例:使用python的多处理库编写分布式代码。使用pool方法创建进程池:
>>> import multiprocessing as MP
>>> # create a process pool:
>>> ppool = MP.Pool()
Pool有一个map方法,但它只接受一个可迭代对象,所以如果你需要传入一个具有更长的形参列表的函数,请将该函数重新定义为partial,以修复除一个以外的所有函数:
>>> ppool.map(pfnx, [4, 6, 7, 8])