Python实现累加函数

发布时间：2019-09-15 09:56:12编辑：auto阅读（2663）

    这个需求比较奇怪，要求实现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

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

关键字：

上一篇： python shell 中文乱码解决

下一篇： python web自动化测试，入门篇