偏函数(partial)和函数柯里化(currying)是函数式编程中常用的技术。有时候我们在复用已有函数时可能需要固定其中的部分参数,这除了可以通过默认值参数来实现之外,还可以使用偏函数。例如有个函数用来实现3个数字相加:
def add3(a, b, c):
return a+b+c
如果现在需要一个类似的函数,与上面的函数add3()的区别仅在于参数b固定为一个数字(例如666),这时就可以使用偏函数的技术来复用上面的函数,例如:
def add2(a, c):
return add3(a, 666, c)
print(add2(1, 1))
或者使用标准库functools提供的partial方法:
from functools import partial
add2 = partial(add3, b=666)
print(add2(a=1, c=1))
函数柯里化除了可以实现偏函数类似的功能之外,还可以利用单参数函数来实现多参数函数,这要归功于Python对函数嵌套定义和lambda表达式的支持。例如:
def func(a):
return lambda b: a+b
print(func(3)(5))
或者
def func(a):
def funcNested(b):
return a+b
return funcNested
print(func(3)(5))
当然,也可以多级嵌套定义函数实现更多参数的需求
def func(a):
def funcNested(b):
def funcNestedNested(c):
return a+b+c
return funcNestedNested
return funcNested
print(func(3)(5)(8))