使用列表推导式实现数字各位求和函数时遇ValueError报错求助

错误原因 #

你的报错根源是自定义函数名和Python内置函数sum重名。从报错栈信息能看到，你实际定义的函数是：

def sum(a):
  return sum([int(i) for i in str(a)])

当函数内部调用sum(...)时，它不会调用内置的求和函数，而是递归调用你自己写的sum函数。此时传入的参数是列表（比如[5,4,1,6,4,6]），转成字符串后会变成"[5, 4, 1, 6, 4, 6]"，遍历这个字符串时会遇到'['字符，尝试把它转为整数就触发了ValueError。

修复方案 #

把自定义函数的名字改成和内置函数不冲突的名称，比如你最初写的add_sum，修正后的代码如下：

# 计算数字各位之和的函数
def add_sum(a):
    return sum([int(i) for i in str(a)])

# 调用函数
result = add_sum(541646)
print(result)  # 输出：26

额外优化 #

列表推导式外层的方括号可以去掉，直接使用生成器表达式，能节省内存开销：

def add_sum(a):
    return sum(int(i) for i in str(a))

内容的提问来源于stack exchange，提问作者Anish Melvin