首页 > 解决方案 > 嵌套列表的元素总和

问题描述

我有一个名为 list_6 的嵌套列表:

[[-1, 1, -1, 1, -1, 1, 1, 0, -1, 0, -1, -1, 1, 1, 1, 0, 1, -1, -1, -1 , 1, 1, 1, 0, 0, -1, 0, 0, 0, 1, 0, -1, 1, 1, -1, 0, 0, 1, 1, -1, 0, -1, 1, 1, -1, 1, -1, -1, -1, 1, -1],...]]

列表的每个元素都包含从 -1,1 开始的整数,并且元素的长度不等,列表中有 20 个元素

我想创建一个名为 list_7 的新列表,如下所示:

[[13],[4],[5],...]],使嵌套列表中的每个元素相加,并打印结果。我尝试使用 iter.zip_longest:

[sum(i) for i in itertools.zip_longest(*list_6, fillvalue=0)]

但我得到一个错误功能:

'str' 对象不可调用

标签: pythonsumnested-listselementwise-operationsstrerror

解决方案


您可以使用列表理解来做到这一点https://www.programiz.com/python-programming/list-comprehension

list_7 = [sum(inner_list) for inner_list in list_6]

在方括号 ( []) 中,您正在遍历 list_6 中的每个项目。由于 list_6 中的每个项目本身都是一个列表 ( inner_list),我们可以在列表中调用 python 的sum函数来获取inner_list https://www.w3schools.com/python/ref_func_sum.asp中的值的总和。

我现在看到您正在寻找要作为列表本身的总和 ( [[13],[4],[5],...]],),在这种情况下您会想要这样做:

list_7 = [[sum(inner_list)] for inner_list in list_6]

将括号括起来sum(inner_list)创建一个新列表,其唯一条目是 inner_list 的总和。


推荐阅读