我如何总结最简洁的Python中的多维数组？

Question

最近的是这个summing columns。

所以我会做同样的事情在我的问题：

说我下面一个Python 2D名单：

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

我可以用一个列表理解的行总计：

row_totals = [ sum(x) for x in my_list ] 

在一行中，我如何求和整个2d数组？

27 

Answer 1

你可以做那么容易，因为

sum(map(sum, my_list)) 

或可替代

sum(sum(x) for x in my_list)) 

收工吧，如果你不希望超过2个维度。请注意，由于使用了map()，第一种解决方案很可能不是最快的（如执行时间）解决方案。基准并根据需要进行比较。

最后，如果您发现自己使用多维数组，请考虑使用NumPy及其优越的数组友好函数。这里是你的问题的简短摘录：

import numpy as np 

my_list = np.array([[1,2,3,4], [2,4,5,6]]) 
np.sum(my_list) 

这将适用于您的数组可能有的任何数量的维度。

Answer 2

>>> sum ([ sum(x) for x in [[1,2,3,4], [2,4,5,6]] ]) 
27 

Answer 3

使用itertools另一种解决方案：

>>> from itertools import chain 
>>> my_list = [ [1,2,3,4], [2,4,5,6] ] 
>>> sum(chain(*my_list)) 
27 

Answer 4

>>> from itertools import chain 
>>> my_list = [[1,2,3,4], [2,4,5,6]] 
>>> sum(chain.from_iterable(my_list)) 
27 

Answer 5

您可以使用和先添加内部列表一起再总结产生的扁平列表：

>>> my_list = [ [1,2,3,4], [2,4,5,6] ] 

>>> sum(my_list, []) 
[1, 2, 3, 4, 2, 4, 5, 6] 

>>> sum(sum(my_list, [])) 
27