Python使用functools模块中的partial函数生成偏函数

1185次阅读  |  发布于5年以前

python 中提供一种用于对函数固定属性的函数(与数学上的偏函数不一样)


    # 通常会返回10进制
    int('12345')  # print 12345 

    # 使用参数 返回 8进制
    int('11111', 8)  # print 4681

每次都得添加参数比较麻烦, functools提供了partial的方法


    import functools

    foo = functools.partial(int, base=8)

    foo('11111')  # print 4681

通过这种方法生成一个固定参数的新函数.

假设要转换大量的二进制字符串,每次都传入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

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

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


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

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


    int2 = functools.partial(int, base=2)

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


    int2('10010')

相当于:


    kw = { 'base': 2 }
    int('10010', **kw)

当传入:


    max2 = functools.partial(max, 10)

实际上会把10作为*args的一部分自动加到左边,也就是:


    max2(5, 6, 7)

相当于:


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

结果为10。

Copyright© 2013-2020

All Rights Reserved 京ICP备2023019179号-8