Question

我在Python3中有以下两个列表：

list1 = [(3, 2), (1, 5), (4, 2), (2, 0)]
list2 = [(1, 6), (2, 1), (3, 3), (4, 0)]

列表中的每个元组都具有（id，score）形式。我想将两个列表中每个ID的得分分开组合，然后按降序排序。因此，输出应为：

list3 = [(1, 11), (3, 5), (4, 2), (2, 1)]

最简单的方法是什么？

Answer 1

使用groupby按ID分组，找到他们的总分，并按降序排序：

from itertools import groupby

list1 = [(3, 2), (1, 5), (4, 2), (2, 0)]
list2 = [(1, 6), (2, 1), (3, 3), (4, 0)]

f = lambda x: x[0]
list3 = sorted([(k, sum(x[1] for x in g)) for k, g in groupby(sorted(list1 + list2, key=f), key=f)], key=lambda x: x[1], reverse=True)

# [(1, 11), (3, 5), (4, 2), (2, 1)]

Answer 2

一个没有任何外部依赖的简单解决方案可能是：

list1 = [(3, 2), (1, 5), (4, 2), (2, 0)]
list2 = [(1, 6), (2, 1), (3, 3), (4, 0)]

list1.sort()
list2.sort()

list3 = [(list1[i][0], list1[i][1] + list2[i][1]) for i in range(len(list1))]
list3.sort(key=lambda x: x[1], reverse=True)

print(list3)

#[(1, 11), (3, 5), (4, 2), (2, 1)]

尽管这假设了一些事情：

您没有重复的ID
您始终在[0]索引中拥有ID，并在[1]中获得分数
您的两个列表总是形状相同
您只有两个列表
您没有丢失的ID（例如，列表1中的ID为1-4，列表2中的ID为2-5）

在这种情况下也许可以吗？取决于广泛适用于您想要的。

Answer 3

一种方法是将至少一个列表转换为字典：

d = dict(list2)
print(sorted(((k, v+d[k]) for k, v in list1), 
             key=lambda a: a[1], reverse=True))

在更一般的情况下，您可能有两个以上的列表，并且并非每个ID都不一定出现在每个列表中，这可以变成：

dicts = [dict(L) for L in [list1, list2]]
keys = set().union(*dicts)   # get all of the IDs
sums = ((k, sum(d.get(k, 0) for d in dicts)) for k in keys)
print(sorted(sums, key=lambda a: a[1], reverse=True))

或更可读

sums = {}
for scores in [list1, list2]:
    for id_, score in scores:`
        if id_ in sums:
            sums[id_] += score
        else:
            sums[id_] = score
 print(sorted(sums, key=lambda a: a[1], reverse=True))

也许最好的选择是使用具有Counter方法的most_common来进行所需的排序。

from collections import Counter
print((Counter(dict(list1)) + Counter(dict(list2))).most_common())

Answer 4

使用 dict理解的另一种解决方案：

sorted({id: dict(list1).get(id, 0) + dict(list2).get(id, 0) for id in dict(list1)}.items(), reverse=True, key=lambda x: x[1])
[(1, 11), (3, 5), (4, 2), (2, 1)]

Python将元组的第二个元素相加

4 个答案: