在 Python 中设定默认参数的另一种办法

在使用一些科学计算的库时，我们会发现他们动不动就十几二十个参数。这些参数太多了，以至于有一些参数我们甚至根本不会修改，但是又不得不添加上去。

如果是我们自己定义的函数，那么可以使用默认参数来实现，例如：

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。

- EOF -

推荐阅读  点击标题可跳转

觉得本文对你有帮助？请分享给更多人

推荐关注「Python开发者」，提升Python技能

点赞和在看就是最大的支持❤️