在使用一些科学计算的库时,我们会发现他们动不动就十几二十个参数。这些参数太多了,以至于有一些参数我们甚至根本不会修改,但是又不得不添加上去。
如果是我们自己定义的函数,那么可以使用默认参数来实现,例如:
def calc(a, b, c, d, e=12, f='test', g=False): s = a + b + c + d * 12 if f: s = s ** 2 if not g: return s else: return s / 2 calc(1, 2, 3, 4)
在调用的时候,由于 e
f
g
都有默认参数,所以我们可以不用传入。
现在问题来了,你调用的是别人已经定义好的函数,假设它有7个参数,但是你只需要修改第3,4个参数。而第一个参数始终固定是1,第二个参数始终是2,此时有没有什么简单的写法呢?
这个时候就可以使用Python的 partial
函数了。
我们使用partial先把原来的函数包装起来,并设定好前两个参数:
from functools import partial simple_calc = partial(calc, 1, 2) data = simple_calc(3, 4)
当我们使用 partial(calc,1,2)
时,它就提前为calc函数设定好了第一、第二个参数分别为1和2,然后当我们再调用 simple_calc
并传入3,4两个参数时,它自动会把1,2,3,4传给calc。这样就实现了,虽然和直接调用 calc(1,2,3,4)
的效果一样,但是我们在后面写代码的时候都只需要传入两个参数,大大简化的代码。
partial也不一定非要按顺序传入参数,也可以通过参数名单独指定后面的参数。例如:
simple_calc = partial(calc, 1, f='test', g=True)
此时就指定了第1个参数为1,名为f的参数的值为test,名为g的参数的值为True。