元组合并列表

创建一个分组字典,其中键是元组中的第一个值,值是列表。然后,您可以将此词典转换为最终列表:

lst = [
    ("1", "a"),
    ("2", "b"),
    ("3", "c"),
    ("3", "d"),
    ("4", "e"),
    ("4", "f"),
    ("5", "g"),
]

out = {}
for a, b in lst:
    out.setdefault(a, []).append(b)

out = list(out.items())
print(out)

打印:

[('1', ['a']), ('2', ['b']), ('3', ['c', 'd']), ('4', ['e', 'f']), ('5', ['g'])]

这是由完成的操作 itertools.groupby 。一个简单的例子,假设您的列表已经正确排序,看起来像

>>> from itertools import groupby
>>> L = [('1', 'a'), ('2', 'b'), ('3', 'c'), ('3', 'd'), ('4', 'e'), ('4', 'f'), ('5', 'g')]
>>> [(k, [x for _, x in v]) for k, v in groupby(L, key=lambda x: x[0])]
[('1', ['a']), ('2', ['b']), ('3', ['c', 'd']), ('4', ['e', 'f']), ('5', ['g'])]

的结果 groupby 是键的映射(由 key 参数)分配给共享每个键的列表中的对象。 [x for _, x in v] 提取原始元组的非关键部分。

以下是实现所需输出的另一种方法

lst = [('1', 'a'), ('2', 'b'), ('3', 'c'), ('3', 'd'), ('4', 'e'), ('4', 'f'), ('5', 'g')]

result = {}

for number, letter in lst:
    if number not in result:
        result[number] = [letter]
    else:
        result[number].append(letter)

result_list = list(result.items())

print(result_list)

1. 创建空字典(结果)
2. 如果字典中还没有数字,它会初始化一个空列表并添加当前字母。如果数字已经存在,它会将字母附加到现有列表中。
3. 最后将结果字典转换为元组