计算包含python中字符串和整数的嵌套列表的总和

Question

我知道如何获得仅包含数字的嵌套列表的总和，但我如何获得同时包含字符串和数字的嵌套列表的总和？

def sum_all_numbers(seq):
    if not seq:
        return 0
    elif isinstance(seq[0], str):
        seq.remove(seq[0])
    elif isinstance(seq[0], list):
        return sum_all_numbers(seq[0]) + sum_all_numbers(seq[1:])
    else:
        return seq[0] + sum_all_numbers(seq[1:])

我想使用双重递归来解决这个问题，但我没有设法让 isinstance 部分正确。

sum_all_numbers(["a", "b", ["c", 3, 4, [5, "d"], 10], 9])

是输入

Answer 1

您可以将所有列表展平为单个列表，然后遍历并过滤列表 reuslt 或在遍历列表时过滤值。

使用第二种方法添加代码

# your code goes here
def get_sum(array: list):
    sum_ = 0
    for i in array:
        if type(i)==int:
            sum_+= i
        elif type(i) in [list, tuple]:
            sum_+= get_sum(i)
    return sum_


result = get_sum(["a", "b", ["c", 3, 4,(1,2,['a', 'b', [1,2,0]],'a',33), [5, "d"], 10], 9])
print(result)
# output 70

Answer 2

我建议使用：

• try/except 尝试加号；
• isinstance(x, (str, bytes)) 测试x 是否为字符串。

这样，您可以添加任何数字，而不仅仅是int。

def sum_all_numbers(seq):
    result = 0
    for x in seq:
        try:
            result += x
        except TypeError:
            if not isinstance(x, (str, bytes)):
                result += sum_all_numbers(x)
    return result

print( sum_all_numbers(["a", "b", ["c", 3, 4, [5, "d"], 10], 9]) )
# 31

请注意，如果嵌套结构包含非数字、字符串或可迭代的内容，则此版本将失败：

sum_all_numbers([None, 3, 4])
# TypeError: 'NoneType' object is not iterable

要修复它，我们可以使用新的try/except 和iter：

def sum_all_numbers(seq):
    result = 0
    for x in seq:
        try:
            result += x
        except TypeError:
            if not isinstance(x, (str, bytes)):
                try:
                    subseq = iter(x)
                    result += sum_all_numbers(subseq)
                except TypeError:
                    pass
    return result

print( sum_all_numbers(["a", "b", ["c", 3, 4, [None, 12.0], [5, "d"], 10], 9]) )
# 43.0

我建议阅读the python sourcecode of more_itertools.collapse，以获得更多想法和错误处理以包含在此功能中。