Question

我们有两个长度可能不同的元组列表，如下所示：

list1 = [(15339456, 140), (15340320, 412), (15341184, 364), (15342048, 488),
         (15342912, 272), (15343776, 350), (15344640, 301), (15345504, 159),
         (15346368, 224), (15347232, 241), (15348096, 223), (15348960, 175)]


list2 = [(15339456, 1516), (15341184, 2046), (15342048, 2400), (15342912, 8370),
         (15343776, 2112), (15344640, 1441), (15345504, 784),  (15346368, 1391)]

每个元组的第一个元素是键，在每个列表中都是唯一的。我们不能假设密钥在两个列表中都存在。一个列表可以包含键不在另一个列表中的元素。现在我们想要总结元组的第二个值，如果它的键在两个列表中，否则我们取完整的元组。

结果：

[(15339456, 1656),
 (15340320, 412),
 (15341184, 2410),
 ...
]

通常使用 zip 对列表进行总结，例如：

for tup1, tup2 in zip(list1, list2):
    sum_ = tup1[1] + tup2[1]
    lst.append((tup1[0], sum_))

如果两个列表的长度相同并且每个键都存在于两个列表中，那么这会起作用，但事实并非如此。

在这个 for 循环中是否有一种很好的方法来构建条件？或者可能是关于这个的pythonic解决方案？两个 for 循环和元素明智的比较似乎不太令人满意。

Answer 1

一个明显的解决方案是创建一个结果字典，然后添加第一个列表中的所有值，然后添加第二个列表中的所有值：

from collections import defaultdict

result = defaultdict(int)
for key, value in list1:
    result[key] += value
for key, value in list2:
    result[key] += value

# convert dictionary-like to list of tuples if you want
result = list(result.items())

使用字典作为结果使您无需进行线性搜索来查找要向其添加值的键（导致整体二次复杂性），尤其是 defaultdict 使您无需这样做

if key not in result:
    result[key] = 0

在添加第一个值之前初始化结果。

您可以使用 itertools.chain 将其推广到任意数量的输入列表：

from collections import defaultdict
from itertools import chain

input_lists = [list1, list2]

result = defaultdict(int)
for key, value in chain.from_iterable(input_lists):
    result[key] += value

从视觉上看，现在只有一个 for 循环，但实际上它也在做同样的事情。

在条件下总结两个元组列表

1 个答案: