给定一个列表字典
vd = {'A': [1,0,1], 'B':[-1,0,1], 'C':[0,1,1]}
我想添加列表元素。所以我想把列表A的第一个元素添加到列表B的第一个,反之亦然复杂的是,你不能依赖标签是A、B、C。它可以是任何东西。其次,字典的长度也是可变的。这里是3。但可能是30。
我需要的结果是一个列表[0,1,3]
所以你只想把所有的值元素相加?
[sum(l) for l in zip(*vd.values())]
简而言之:
>>> map(sum, zip(*vd.values()))
[0, 1, 3]
解释
给定一本字典:
>>> vd = {'A': [1,0,1], 'B': [-1,0,1], 'C': [0,1,1]}
我们可以得到值:
>>> values = vd.values()
>>> values
[[1, 0, 1], [-1, 0, 1], [0, 1, 1]]
然后把它们拉上拉链:
>>> zipped = zip(*values)
>>> zipped
[(1, -1, 0), (0, 0, 1), (1, 1, 1)]
注意,zip
压缩每个参数;它不需要一个东西的清单来压缩。因此,我们需要*
将列表解压缩为参数。
如果我们只有一个列表,我们可以将它们相加:
>>> sum([1, 2, 3])
6
然而,我们有多个,所以我们可以在上面进行映射:
>>> map(sum, zipped)
[0, 1, 3]
总之:
>>> map(sum, zip(*vd.values()))
[0, 1, 3]
延伸到平均值而不是总和
这种方法也很容易扩展;例如,我们可以很容易地使其对元素求平均值,而不是求和。要做到这一点,我们首先要制作一个average
函数:
def average(numbers):
# We have to do the float(...) so it doesn't do an integer division.
# In Python 3, it is not necessary.
return sum(numbers) / float(len(numbers))
然后将sum
替换为average
:
>>> map(average, zip(*vd.values()))
[0.0, 0.3333, 1.0]
vd = {'A': [1,0,1], 'B':[-1,0,1], 'C':[0,1,1]}
vd_keys = list(vd.keys())
rt = vd[vd_keys.pop()].copy() # copy otherwise rt and vd[vd_keys.pop()] will get synced
for k in vd_keys:
for i in range(len(rt)):
rt[i] += vd[k][i]
print(rt)