使用相同的键合并元组

Question

如何使用相同的键合并元组

list_1 = [("AAA", [123]), ("AAA", [456]), ("AAW", [147]), ("AAW", [124])]

并将它们变成

list_2 = [("AAA", [123, 456]), ("AAW", [147, 124])]

Answer 1

最高效的方法是使用collections.defaultdict字典将数据存储为扩展列表，然后根据需要转换回元组/列表：

import collections

list_1 = [("AAA", [123]), ("AAA", [456]), ("AAW", [147]), ("AAW", [124])]

c = collections.defaultdict(list)
for a,b in list_1:
    c[a].extend(b)  # add to existing list or create a new one

list_2 = list(c.items())

结果：

[('AAW', [147, 124]), ('AAA', [123, 456])]

请注意，转换后的数据可能最好留作字典。 再次转换为列表会丢失字典的“关键”功能。

另一方面，如果你想保留原始元组列表的“键”的顺序，除非你使用python 3.6 / 3.7，你必须创建一个包含原始“键”的列表（有序） ，unique），然后从字典重建列表。 或使用OrderedDict然后你不能使用defaultdict （或使用食谱 ）

Answer 2

您可以使用dict来跟踪每个键的索引以保持时间复杂度O（n）：

list_1 = [("AAA", [123]), ("AAA", [456]), ("AAW", [147]), ("AAW", [124])]
list_2 = []
i = {}
for k, s in list_1:
    if k not in i:
        list_2.append((k, s))
        i[k] = len(i)
    else:
        list_2[i[k]][1].extend(s)

list_2将成为：

[('AAA', [123, 456]), ('AAW', [147, 124])]

Answer 3

您可以创建字典并循环遍历列表。 如果字典中的项目将值附加到现有列表，则将值分配给键。

dict_1 = {}
for item in list_1:
    if item[0] in dict_1:
        dict_1[item[0]].append(item[1][0])
    else:
        dict_1[item[0]] = item[1]
list_2 = list(dict_1.items())

Answer 4

与其他答案类似，您可以使用字典将每个键与值列表相关联。 这是在下面的代码片段中的函数merge_by_keys中实现的。

import pprint

list_1 = [("AAA", [123]), ("AAA", [456]), ("AAW", [147]), ("AAW", [124])]

def merge_by_key(ts):

    d = {}
    for t in ts:
        key = t[0]
        values = t[1]
        if key not in d:
            d[key] = values[:]
        else:
            d[key].extend(values)

    return d.items()



result = merge_by_key(list_1)

pprint.pprint(result)