Python 对元组列表的列表进行分组

发布于08月02日

我有一个元组列表，例如:

lst = [[1, 4, (1, 1, 1, 4)], [7,8,(2, 1,6, 14)],[2,3, (1, 1, 1, 9)], [5,4, (2, 1,1, 12)], [8,9, (2, 1,1, 99)],  [5,7,(2, 1,6, 19)], [3,4, (1, 1, 1, 14)]]

我需要根据元组对列表进行分组，以便所有列表都在同一组中，只有元组中的最后一项不同.例如:

first group: [[1, 4, (1, 1, 1, 4)], [2,3, (1, 1, 1, 9)], [3,4, (1, 1, 1, 14)]]
second group: [[5,4, (2, 1,1, 12)], [8,9, (2, 1,1, 99)]]
third group: [[7,8,(2, 1,6, 14)], [5,7,(2, 1,6, 19)]]

推荐答案

try :

lst = [
    [1, 4, (1, 1, 1, 4)],
    [7, 8, (2, 1, 6, 14)],
    [2, 3, (1, 1, 1, 9)],
    [5, 4, (2, 1, 1, 12)],
    [8, 9, (2, 1, 1, 99)],
    [5, 7, (2, 1, 6, 19)],
    [3, 4, (1, 1, 1, 14)],
]

out = {}
for i in lst:
    *key, _ = i[-1]
    out.setdefault(tuple(key), []).append(i)

print(list(out.values()))

打印:

[
    [[1, 4, (1, 1, 1, 4)], [2, 3, (1, 1, 1, 9)], [3, 4, (1, 1, 1, 14)]],
    [[7, 8, (2, 1, 6, 14)], [5, 7, (2, 1, 6, 19)]],
    [[5, 4, (2, 1, 1, 12)], [8, 9, (2, 1, 1, 99)]],
]