python - 查找嵌套整数列表的总和

Question

import math
lists =  [1,[2,3],4]
total = 0
for i in range(len(lists)):
    total += sum(i)
print(total)

我要打印，

>>>10

但是抛出一个错误。

我想让它把所有数字相加，包括嵌套 if 中的数字。

Answer 1

在您的程序中，for i in range(len(lists)) - 计算结果为 3，因为 lists 对象有 3 个元素。在循环 total += sum(i) 中，它会尝试执行 int + list 操作，这会导致错误。因此，您需要检查类型，然后添加各个元素。

def list_sum(L):
    total = 0  
    for i in L:
        if isinstance(i, list): 
            total += list_sum(i)
        else:
            total += i
    return total

这是@pavelanossov 的评论——以更优雅的方式做同样的事情

sum(sum(i) if isinstance(i, list) else i for i in L)