python - 对数组的每个元素求和

Question

这个问题已经有答案了:

How would I sum a multi-dimensional array in the most succinct python? (5 个回答)

已关闭 7 年前。

我有一个数组

[array([  2.46000000e-02,   4.28211000e+01,   7.95400000e+00]), array([  0.0987,  15.1704,   7.5235]), array([ 0.2502,  7.5852,  0.041 ])]

我正在查找数组行中项目的总和对于 acb 中的项目，上面的数组:

for item in acb:
   print    sum(item) 



50.7997
22.7926
7.8764

我想生成一个包含以上数字 50.799,22.79,7.87 的数组

谢谢

Answer 1

如果您没有将数组作为多维 numpy 数组，您可以使用 np.sum() 来获取第二个轴上的项目总和:

>>> A = [np.array([  2.46000000e-02,   4.28211000e+01,   7.95400000e+00]), np.array([  0.0987,  15.1704,   7.5235]), np.array([ 0.2502,  7.5852,  0.041 ])]
>>> A = np.array(A)
>>> 
>>> np.sum(A, axis=1)
array([ 50.7997,  22.7926,   7.8764])

请注意，如果您正在处理大型数据集，这会更有效，否则您可以简单地使用 map() 函数来获取列表中所有数组的总和(在 python3 中是迭代器) )。