当函数的参数个数太多,需要简化时,使用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 (偏函数)