如何在最简洁的 python 中对多维数组求和?
问题描述
最接近的是这个汇总列.
所以我会在我的问题中做类似的事情:
So I'll do something similar in my question:
假设我有一个 Python 2D 列表,如下所示:
Say I've a Python 2D list as below:
my_list = [ [1,2,3,4],
[2,4,5,6] ]
我可以通过列表理解获得行总数:
I can get the row totals with a list comprehension:
row_totals = [ sum(x) for x in my_list ]
如何在一行中求和整个二维数组?
In one line, how can I sum the entire 2d-array?
27
解决方案
你可以做到这么简单
sum(map(sum, my_list))
或者
sum(sum(x) for x in my_list))
如果您不期望超过 2 个维度,那就收工吧.请注意,由于 map()
的使用,第一个解决方案很可能不是最快的(如执行时间)解决方案.必要时进行基准测试和比较.
and call it a day, if you don't expect more than 2 dimensions. Note that the first solution is most likely not the fastest (as in execution time) solution, due to the usage of map()
. Benchmark and compare as necessary.
最后,如果您发现自己使用多维数组,请考虑使用 NumPy 及其优越的数组友好函数.以下是您的问题的简短摘录:
Finally, if you find yourself using multidimensional arrays, consider using NumPy and its superior array-friendly functions. Here's a short excerpt for your problem:
import numpy as np
my_list = np.array([[1,2,3,4], [2,4,5,6]])
np.sum(my_list)
这适用于您的数组可能具有的任意数量的维度.
This would work for any number of dimensions your arrays might have.
相关文章