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程序员展示一个常见的反模式，不循环索引，循环遍历对象本身。 E.g：

for item in b:
    do_something(item)

而不是：

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

更清晰，更简单，更快捷。

那就是说，你遇到的主要问题是其中一个项目不是一个列表，所以它没有长度。

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

>>> 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()不会处理不可迭代的项目，因此我们将任何内容放入列表中以解决此问题。

另一种方法是制作自己的发电机：

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

尝试这样的事情，仅在元素为sum()或list时才使用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 3

基于生成器的解决方案

>>> 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 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)

添加所有嵌套列表中的元素

5 个答案:

另一种方法