functools 之 partial(偏函数)

当函数的参数个数太多,需要简化时,使用functools.partial可以创建一个新的函数,这个新函数可以固定住原函数的部分参数,从而在调用时更简单。当然,decorator(装饰器) 也可以实现,如果,我们嫌麻烦的话。

我们借助Python的help帮助函数,简单了解下什么是partial(偏函数):

这里我们主要说下紫色圈的意思:

partial 一共有三个部分:

(1)第一部分也就是第一个参数,是一个函数,这个函数可以是你定义的,也可以是Python内置函数

(2)第二部分是一个可变参数,*args,比如内置函数max的参数就是一个可变参数,max(1,2,3,4,5)=5

 

(3)第三部分是一个关键字参数,比如内置函数int的第二个参数就是命名关键字参数,默认base=10,表示int转换时默认是10进制的:

 

所以:

  partial函数的作用就是:将所作用的函数作为 partial() 函数的第一个参数,原函数的各个参数依次作为 partial() 函数的后续参数,原函数有关键字参数的一定要带上关键字,没有的话,按原有参数顺序进行补充。

举例说明:

1. 参数一层一层传

from functools import partial


def my_sum(a, b):
    print("a=", a)
    print("b=", b)
    return a + b


# 偏函数的第一个参数是 作用函数(my_sum)
# 偏函数的第二个参数是 传给作用函数的第一个参数(a)
# 偏函数的第三个参数是 传给作用函数的第三个参数(b)
new_my_sum = partial(my_sum, 10)        # 相当于: new_my_sum = my_sum(10, b)   位置传参轮到a了, a=10
res = new_my_sum(20)                    # 相当于: res = my_sum(10, 20)  位置传参轮到b了, b=20

print("res=", res)

# 结果:
# a= 10
# b= 20
# res= 30

2. 参数一次性传进去

from functools import partial


def my_sum(a, b):
    print("a=", a)
    print("b=", b)
    return a + b


# 偏函数的第一个参数是 作用函数(my_sum)
# 偏函数的第二个参数是 传给作用函数的第一个参数(a)
# 偏函数的第三个参数是 传给作用函数的第三个参数(b)
new_my_sum = partial(my_sum, 10, 20)        # 相当于: new_my_sum = my_sum(10, b)   位置传参轮到a了, a=10; 轮到b了, b=20
res = new_my_sum()                    # 相当于: res = my_sum(10, 20)

print("res=", res)

# 结果:
# a= 10
# b= 20
# res= 30

3. 关键字传参(1)

from functools import partial


def my_sum(a, b):
    print("a=", a)
    print("b=", b)
    return a + b


# 偏函数的第一个参数是 作用函数(my_sum)
# 偏函数的第二个参数是 传给作用函数的第一个参数(a)
# 偏函数的第三个参数是 传给作用函数的第三个参数(b)

# 注意: 这里的关键字传参,传给的是后边的b,前边的a未指定关键字,没有报错,正常,因为按照形参a和b的位置来看,关键字参数要在位置参数之后
new_my_sum = partial(my_sum, b=10)        # 相当于: new_my_sum = my_sum(a, 10)   关键字传参,传给了b, b=10
res = new_my_sum(20)                      # 相当于: res = my_sum(20, 10)         b有了实参,剩下a,所以传给了a, a=20

print("res=", res)

# 结果:
# a= 20
# b= 10
# res= 30

4. 关键字传参(2)

from functools import partial


def my_sum(a, b):
    print("a=", a)
    print("b=", b)
    return a + b


# 偏函数的第一个参数是 作用函数(my_sum)
# 偏函数的第二个参数是 传给作用函数的第一个参数(a)
# 偏函数的第三个参数是 传给作用函数的第三个参数(b)

# 注意: 这里的关键字传参,传给的是前边的a,后边的b未指定关键字,就会报错,因为按照形参a和b的位置来看,关键字参数要在位置参数之后,也就是说,a指定了,b也必须指定就可以了
new_my_sum = partial(my_sum, a=10)        # 相当于: new_my_sum = my_sum(10, b)   关键字传参,传给了a, a=10
res = new_my_sum(20)                      # 相当于: res = my_sum(10, 20)         a有了实参,剩下b,所以想要传给b, b=20,但是违反了关键字参数在位置参数之后原则,所以报错

print("res=", res)

# 结果:
# TypeError: my_sum() got multiple values for argument 'a'

本文参考:

  https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/00143184474383175eeea92a8b0439fab7b392a8a32f8fa000  (偏函数和装饰器的对比)

  https://blog.csdn.net/appleyk/article/details/77609114  (偏函数)

猜你喜欢

转载自www.cnblogs.com/yuanyongqiang/p/10453911.html