欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

对Python实现累加函数的方法详解

程序员文章站 2023-11-16 12:29:40
这个需求比较奇怪,要求实现sum和magasum函数,实现以下功能 sum(1) =>1 sum(1,2,3) =>6 megasum(1)()...

这个需求比较奇怪,要求实现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实现累加函数的方法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。