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

原创文章,作者:kirin,如若转载,请注明出处:https://blog.ytso.com/219861.html

(0)
上一篇 2022年1月2日
下一篇 2022年1月2日

相关推荐

发表回复

登录后才能评论