首页 > 编程笔记

Python偏函数的理解

Python 的 functools 模块提供了很多有用的功能,其中一个就是偏函数(Partial Function)。

注意,这里的偏函数和数学意义上的偏函数不一样。

通过设置参数的默认值,可以降低函数调用的难度,偏函数也可以做到这一点。

例如,int() 函数可以把字符串转换为整数,当仅传入字符串时,int() 函数默认按十进制转换:
>>>print(int('2888'))
2888
int() 函数还提供了 base 参数,默认值为 10。如果传入 base 参数,就可以进行 N 进制的转换:
>>>print(int('123456', base=8))
42798
>>>print(int('123456', 16))  #base也可以省略,直接传入base的值
1193046
假设要转换大量的二进制字符串,而每次都传入 int(x, base=2) 就会非常麻烦,这里可以定义一个 int2() 函数,默认把 base=2 传进去:
>>>def int2(x, base=2):
...    return int(x, base)
...
>>>print(int2('1001000'))
72
>>>print(int2('1000011'))
67
>>>print(int2('1001110'))
78
functools.partial 就是帮助用户创建偏函数的,不需要再自定义 int2() 函数,可以直接使用下面的代码创建一个新的函数 int2:
>>>from functools import partial
>>>int2 = partial(int, base=2)
>>>print(int2('1001000'))
72
>>>print(int2('1000011'))
67
>>>print(int2('1001110'))
78
可见,functools.partial 的作用就是把一个函数的某些参数固定住(设置默认值),返回一个新函数,调用这个新函数会更简单。

注意:int2 函数仅仅是把 base 参数的默认值重新设置为 2,也可以在函数调用时传入其他值:
>>>print(int2('1000000', base=10))
1000000
当函数的参数数量太多、需要简化时,使用 functools.partial 可以创建一个新函数,这个新函数可以固定住原函数的部分参数,从而使调用更简单。

推荐阅读