Python 偏函数

Python 偏函数是通过 functools 模块被用户调用。

偏函数 partial 应用

函数在执行时，要带上所有必要的参数进行调用。但是，有时参数可以在函数被调用之前提前获知。这种情况下，一个函数有一个或多个参数预先就能用上，以便函数能用更少的参数进行调用。

偏函数是将所要承载的函数作为partial()函数的第一个参数，原函数的各个参数依次作为partial()函数后续的参数，除非使用关键字参数。

通过语言描述可能无法理解偏函数是怎么使用的，那么就举一个常见的例子来说明。在这个例子里，我们实现了一个取余函数，对于整数 100，取得对于不同数 m 的 100%m 的余数。

from functools import partial
 
def mod( n, m ):
  return n % m
 
mod_by_100 = partial( mod, 100 )
 
print mod( 100, 7 )  # 2
print mod_by_100( 7 )  # 2

由于之前看到的例子一般选择加法或乘法来讲解，无法体会偏函数参数的位置问题，容易给人造成 partial 的第二个参数也是原函数的第二个参数的假象，所以我在这里选择 mod 来讲解。

而对于有关键字参数的情况下，就可以不按照原函数的参数位置和个数了。下面再看一个例子，讲的是如何进行不同的进制转换。

from functools import partial
 
bin2dec = partial( int, base=2 )
print bin2dec( '0b10001' )  # 17
print bin2dec( '10001' )  # 17
 
hex2dec = partial( int, base=16 )
print hex2dec( '0x67' )  # 103
print hex2dec( '67' )  # 103

偏函数的这些应用看似简单，用途却很大，可以很好的执行DRY原则，节省编程成本。

补充：

关于偏函数，可以对现有函数进行加强。

>>> import functools
>>> int2 =functools.partial (int, base=2) # 把 int 的转换设为二进制了，这里 base 是 int 函数表示进制的参数。
>>>int2('1000000')
64

>>>int2('1010101')
85

在重设之后, 也可以在函数调用时传入其他值:

​>>> int2('1000000', base=10)  # 这里 base 变成了 10，覆盖了已设的默认值 2。
1000000

注意这里在创建新的偏函数后，依旧可以更改已经设置的默认值，但是必须清晰指出是更改了 base 的值。

否则如果直接传如一个数值会报错:

int2('100', 10)  # 报错，10 前未加 base=，不能分辨这是传给 base 的

原因如下:

创建偏函数时，实际上可以接收函数对象、*args 和 **kw 这 3 个参数，当传入：

int2 =functools.partial(int, base=2)  # int 是函数对象，base=2 是 **kw，没有传入 *args 参数

实际上固定了 int() 函数的关键字参数 base，也就是：

int2('10010')

相当于：

kw = { 'base': 2 }
int('10010', **kw)  # 如果不指明 base=2 而只是传入 2 的话，那么 2 被认为是 *args 的值

当传入：

max2 =functools.partial(max, 10)  # 这里的 10 显然就是作为 *args 里的值传入的

实际上会把 10 作为 *args 的一部分自动加到参数列表里去，也就是：

max2(5,6, 7)  # 原本有 *args 的列表，然后会把 10 再加进去

相当于：

args = (10, 5, 6, 7)
max(*args)

结果为 10。