Python中的偏函数

  • Post author:
  • Post category: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)

这样,我们转换二进制就非常方便了:

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
>>> 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

传入的,实际上是:

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
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可以创建一个新的函数,这个新函数可以固定住原函数的部分参数,从而在调用时更简单。