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

Question

给出列表字典

vd = {'A': [1,0,1], 'B':[-1,0,1], 'C':[0,1,1]}

我想明智地添加列表元素。所以我想将列表A中的第一个元素添加到列表B的第一个元素，反之亦然 复杂性是你不能依赖标签A，B，C。它可以是任何东西。第二，字典的长度也是可变的。这里是3.但它可能是30。

我需要的结果是列表[0,1,3]

Answer 1

所以你只想按元素加起来所有的值？

[sum(l) for l in zip(*vd.values())]

Answer 2

简而言之：

>>> map(sum, zip(*vd.values()))
[0, 1, 3]

---

解释

给出一本字典：

>>> vd = {'A': [1,0,1], 'B': [-1,0,1], 'C': [0,1,1]}

我们可以get the values：

>>> values = vd.values()
>>> values
[[1, 0, 1], [-1, 0, 1], [0, 1, 1]]

然后zip他们：

>>> zipped = zip(*values)
>>> zipped
[(1, -1, 0), (0, 0, 1), (1, 1, 1)]

请注意zip拉上每个参数;它没有列出要拉链的东西。因此，我们需要*将列表解压缩为参数。

如果我们只有一个列表，我们可以sum：

>>> sum([1, 2, 3])
6

但是，我们有多个，所以我们可以map覆盖它：

>>> 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]

Answer 3

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)