如何在python中添加列表dictionary中的元素



给定一个列表字典

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)

最新更新