Python 3.6从入门到精通(视频教学版)
上QQ阅读APP看书,第一时间看更新

6.11 偏函数

Python的functools模块提供了很多有用的功能,其中一个就是偏函数(Partial function)。注意,这里的偏函数和数学意义上的偏函数不一样。

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

例如:

int()函数可以把字符串转换为整数,当仅传入字符串时,int()函数默认按十进制转换:

    >>> int('2888')
    2888

int()函数还提供了base参数,默认值为10。如果传入base参数,就可以进行N进制的转换:

    >>> int('123456', base=8)
    42798
    >>> int('123456', 16)  #base也可以省略,直接传入base的值
    1193046

假设要转换大量的二进制字符串,而每次都传入int(x, base=2)就会非常麻烦,这里可以定义一个int2()函数,默认把base=2传进去:

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

    >>> int2('1001000')
    72
    >>> int2('1000011')
    67
    >>> int2('1001110')
    78

functools.partial就是帮助用户创建偏函数的,不需要再自定义int2()函数,可以直接使用下面的代码创建一个新的函数int2:

    >>> from functools import partial
    >>> int2 = partial(int, base=2)
    >>> int2('1001000')
    72
    >>> int2('1000011')
    67
    >>> int2('1001110')
    78

可见,functools.partial的作用就是把一个函数的某些参数固定住(设置默认值),返回一个新函数,调用这个新函数会更简单。

注意 int2函数仅仅是把base参数的默认值重新设置为2,也可以在函数调用时传入其他值:

    >>> int2('1000000', base=10)
    1000000

当函数的参数数量太多、需要简化时,使用functools.partial可以创建一个新函数,这个新函数可以固定住原函数的部分参数,从而使调用更简单。