Python怎么实现多维列表求和

本篇内容介绍了“Python怎么实现多维列表求和”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！

对于简单的列表，比如[1, 2, 3]，求和很简单，直接sum就可以了。

但是对于多维的，比如[[1, 2, 3], [4, 5, 6], [7], [8, 9]]，该怎么求和呢？

这里DE8UG介绍两种方法，第一种很好想，就是像剥洋葱一样一层层的递归求和。

第二种方法，就需要利用一些Python内部的好东西了。

方法1，递归操作，层层递进来处理

def recursive_list_sum(data_list):
total = 0
for element in data_list:
if type(element) == list:
total = total + recursive_list_sum(element)
else:
total = total + element

return total

方法2，先铺平列表，然后直接求和

这里铺平列表的方式就多了，我们看两个itertools.chain和sum

其中itertools.chain可以用静态方法也可以直接用类的初始化函数来搞定。

sum的小技巧是第一个参数放列表，第二个参数直接写[], 这种方式直接把列表铺平。

import itertools

list2d = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
# a = list(itertools.chain.from_iterable(list2d))  # call staticmethod from_iterable
a = list(itertools.chain(*list2d))   # init an object
b = sum(list2d, [])
print(a, b)

最后在把列表直接sum一下就搞定了。

“Python怎么实现多维列表求和”的内容就介绍到这里了，感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站，小编将为大家输出更多高质量的实用文章！