Python:n列表的元素总和

Question

任何人都可以给我一个提示如何聚合(总结)多个列表的具体元素？

例如,我有20个具有相似名称模式的列表,每个列表代表相应家庭(代理)的消费支出(即指数):

c_agent_0 = [10.0, 11.0, ...]
.
.
.
c_agent_19 = [8.0, 9.0, ...]

我需要一些像value = index of sum of c_agent _*[whatever]

谢谢!

Answer 1

使用zip将"组合"列表合并为单个列表,然后对每个索引求和元素:

c_agent_0 = [10.0, 11.0, 12.0, 13.0, 14.0]
c_agent_1 = [1.1, 12.3, 14.2, 13.2, 14.3]
c_agent_2 = [1.4, 12.1, 14.5, 13.1, 14.2]

zipped_list = zip(c_agent_0, c_agent_1, c_agent_2)

print zipped_list
print [sum(item) for item in zipped_list]

版画

[(10.0, 1.1, 1.4), (11.0, 12.3, 12.1), (12.0, 14.2, 14.5), (13.0, 13.2, 13.1), (14.0, 14.3, 14.2)]

[12.5, 35.4, 40.7, 39.3, 42.5]

Answer 2

您可以创建一个列表列表，然后使用 numpy 数组：

import numpy as np
c_agent_0 = [10.0, 11.0, 12.0]
c_agent_1 = [13.0, 14.0, 15.0]
c_agent_2 = [16.0, 17.0, 18.0]
mylist = [c_agent_0, c_agent_1, c_agent_2]
mylist = sum(map(np.array, mylist))
print mylist[0]

请注意，列表理解很可能更快；这只是 numpy 的一个解决方案:)。