python - 将所有嵌套列表中的元素相加

Question

我正在添加所有嵌套列表中的元素，到目前为止我所拥有的是

for i in range(len(b)) :
    for j in range(len(b[i])):
        total = total + b[i][j]

但是当嵌套列表中存在整数值时，这将不起作用，例如，

b = [[1,2], [3,4], [5,6], 1]

在这种情况下，我收到一条错误消息，指出“int”类型的对象没有 len()。在这种情况下我该怎么办？

Answer 1

首先，您向 Python 程序员展示了一个常见的反模式，不要循环遍历索引，循环遍历对象本身。例如：

for item in b:
    do_something(item)

而不是：

for i in range(len(b)):
    do_something(b[i])

它更清晰、更简单、更快捷。

也就是说，您遇到的主要问题是其中一项不是列表，因此它没有长度。

这里更好的选择是使用生成器表达式 和 展平列表itertools.chain.from_iterable()，然后使用内置sum()函数对元素求和。

>>> import collections
>>> import itertools
>>> b = [[1,2], [3,4], [5,6], 1]
>>> list(itertools.chain.from_iterable(item if isinstance(item, collections.Iterable) else [item] for item in b))
[1, 2, 3, 4, 5, 6, 1]
>>> sum(itertools.chain.from_iterable(item if isinstance(item, collections.Iterable) else [item] for item in b))
22

我们需要生成器表达式，因为它itertools.chain()不会处理不可迭代的项目，因此我们将 any 放入列表中以解决此问题。

另一种方法是制作自己的生成器：

def chain_mixed(iterable):
    for item in iterable:
        try:
            for subitem in item:
                yield subitem
        except TypeError:
            yield item

然后你可以简单地做：

sum(chain_mixed(b))

Answer 2

基于生成器的解决方案

>>> b = [[1,2], [3,4], [5,6], 1]
>>> def sum_up(x):
...     for i in x:
...             if isinstance(i, int):
...                     yield i
...             else:
...                     for j in sum_up(i):
...                             yield j
... 
>>> sum(sum_up(b))
22
>>> b = [[1,2], [3,4], [5,6,[3,[6,7]]], 1]
>>> sum(sum_up(b))
38

Answer 3

尝试这样的事情，sum()仅当元素是listor时tuple使用，否则简单使用数字。

>>> b = [[1,2], [3,4], [5,6], 1]
>>> sum(sum(x) if isinstance(x,(list,tuple)) else x for x in b)
22

改进您的解决方案：

total=0
b = [[1,2], [3,4], [5,6], 1]
for i in range(len(b)) :
    if isinstance(b[i],(list,tuple)):
        for j in range(len(b[i])):
            total += b[i][j]
    else:
        total +=b[i]
print total  #prints 22

Answer 4

一种方法是

def calc_sum(the_list):
    sum = 0
    for item in the_list:
        if isinstance(item,(list,tuple)):
           sum += calc_sum(item)
        else:
           sum += item
    return sum

不是那种pythonic ...但它应该可以工作

Answer 5

另一种方法来做到这一点

如果列表包含嵌套列表和整数/浮点数，这将起作用这里我使用 python 内置函数“sum”来添加列表的所有元素

def add(nested_list):
    temp=[]
    for i in nested_list:
        if isinstance(i,list):
            temp.append(sum(i))
        else:
            temp.append(i)
    print sum(temp)