首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >Python实现累加函数

Python实现累加函数

作者头像
用户7886150
修改于 2021-01-11 10:18:07
修改于 2021-01-11 10:18:07
2.3K0
举报
文章被收录于专栏:bit哲学院bit哲学院

参考链接: Python中的sum函数

这个需求比较奇怪,要求实现Sum和MagaSum函数,实现以下功能 

Sum(1) =>1

Sum(1,2,3) =>6

MegaSum(1)() =>1

MegaSum(1)(2)(3)() =>6 

实际上Sum就是Python自建的sum函数,它支持变参,变参怎么实现,自然是*args,所以很容易写出雏形:  Sum 

def Sum(*args):

    count = 0

    for i in args:

        count+=i

    return count 

第二个函数就有点皮了,它要求有参数的时候,返回的值是一个函数的引用,无参数的时候,返回的是结果。  要实现可以有或无参数,那么首想到的是Python的默认参数。我们可以默认参数值是None,当主动传入参数时,返回一个函数的引用,否则,返回结果。这里用到了Python可变类型作为函数参数时的一些特性。代码如下: 

def Sum(*args):

    s = 0

    for i in args:

        s+=i

    return s

def megaSum(s=None,l=[]):

    if not s:

        ret = Sum(*l)

        l.clear() # 清空,否则之后计算的结果值会不对。

        return ret

    else:

        l.append(s)

        return megaSum

print(Sum(1,2,3))

print(megaSum(1)(2)(3)())

print(megaSum(1)()) 

实际上还是有一些Bug的,原因是只有当不传参数的时候才会清空l的内容,这样就会造成如 

print(megaSum(1)(2)(3)) #<function megaSum at 0x7fb533743a60>

print(megaSum(1)()) #7 

计算结果不准确。这个怎么解决,暂时还没想到,如果有朋友知道可以告诉我。

本文系转载,前往查看

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

问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档