前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >一日一技:在Python中为别人的函数设定默认参数

一日一技:在Python中为别人的函数设定默认参数

作者头像
青南
发布2019-05-17 15:38:48
1.1K0
发布2019-05-17 15:38:48
举报
文章被收录于专栏:未闻Code未闻Code
在使用一些科学计算的库时,我们会发现他们动不动就十几二十个参数。这些参数太多了,以至于有一些参数我们甚至根本不会修改,但是又不得不添加上去。

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

代码语言:javascript
复制
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先把原来的函数包装起来,并设定好前两个参数:

代码语言:javascript
复制
from functools import partialsimple_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也不一定非要按顺序传入参数,也可以通过参数名单独指定后面的参数。例如:

代码语言:javascript
复制
simple_calc = partial(calc, 1, f='test', g=True)

此时就指定了第1个参数为1,名为f的参数的值为test,名为g的参数的值为True。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2019-05-16,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 未闻Code 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档