Python中的偏函数
偏函数(Partial function)是Python的functools模块提供的一个很有用的功能。
简单总结functools.partial
的作用就是,把一个函数的某些参数给固定住(也就是设置默认值),返回一个新的函数,调用这个新函数会更简单。
举个例子,int()
函数可以把字符串转换为整数,当仅传入字符串时,int()
函数默认按十进制转换:
>>> int('12345') 12345
但int()
函数还提供额外的base
参数,默认值为10
。如果传入base
参数,就可以做N进制的转换:
>>> int('12345', base=8) 5349 >>> int('12345', 16) 74565
假设要转换大量的二进制字符串,每次都传入int(x, base=2)
非常麻烦,于是,我们想到,可以定义一个int2()
的函数,默认把base=2
传进去:
def int2(x, base=2): return int(x, base)
这样,我们转换二进制就非常方便了:
>>> int2('1000000') 64 >>> int2('1010101') 85
functools.partial可以
帮助我们创建一个偏函数的,不需要我们自己定义int2()
,可以直接使用下面的代码创建一个新的函数int2
:
>>> import functools >>> int2 = functools.partial(int, base=2) >>> int2('1000000') 64 >>> int2('1010101') 85
注意到上面的新的int2
函数,仅仅是把base
参数重新设定默认值为2
,但也可以在函数调用时传入其他值
>>> int2('1000000', base=10) 1000000
最后,创建偏函数时,实际上可以接收函数对象、*args
和**kw
这3个参数,所以当传入:
int2 = functools.partial(int, base=2) int2('10010')
因为偏函数定义处有base关键字,所以是以**kw传入的,实际上是:
kw = { 'base': 2 } int('10010', **kw)
同理,当传入:
max2 = functools.partial(max, 10)
max2(5, 6, 7)
因为偏函数定义处没有关键字,所以是以*arg传入的,实际上是(把10
作为*args
的一部分自动加到左边)
args = (10, 5, 6, 7)
max(*args)
所以,当函数的参数个数太多,需要简化时,使用functools.partial
可以创建一个新的函数,这个新函数可以固定住原函数的部分参数,从而在调用时更简单。