合并相同元素至列表：嵌套列表按子列表第二个元素分组的实现问题

我来帮你搞定这个列表转换的问题！你的需求是把原列表中第二个元素相同的子项归为一组，每组的结构是[[对应第一个元素的集合], 共同的第二个元素]，最终组合成目标格式对吧？

先说说你现有代码的问题 #

• 嵌套循环的逻辑会重复添加元素：比如i=0时匹配j=2，会把1和3加入列表；i=2时匹配j=5，又会把3和6加入，最终导致元素重复。
• 没有处理仅单个元素的组：比如原列表中的[5,[5,6,7]]，你的代码会完全忽略它。
• 没有把分组后的结果正确整合到最终列表lst中，循环结束后lst还是空的。
• 用列表直接做比较虽然可行，但如果要分组存储，列表不可哈希的特性会限制使用更高效的字典结构。

正确的实现方案 #

最简洁高效的方式是用字典分组，利用字典的键来区分不同的第二个元素，值用来收集对应的第一个元素。代码如下：

alist = [[1,[1,2]],[2,[3,4,5]],[3,[1,2]],[4,[3,4,5]],[5,[5,6,7]],[6,[1,2]]]

# 1. 用字典分组：键是转成元组的第二个元素（列表不可哈希，转元组才能当字典键）
group_dict = {}
for item in alist:
    # 将子列表的第二个元素转为元组，作为字典的键
    key = tuple(item[1])
    # 如果键不存在，初始化一个空列表
    if key not in group_dict:
        group_dict[key] = []
    # 把当前子项的第一个元素加入对应键的列表
    group_dict[key].append(item[0])

# 2. 将字典转换为目标格式的列表
result = []
for key, values in group_dict.items():
    # 如果需要过滤掉只有单个元素的组（和你的目标结果一致），就加这个判断
    if len(values) > 1:
        # 把元组转回列表，组成目标结构
        result.append([values, list(key)])

print(result)

运行后输出就是你想要的目标格式：

[[[1, 3, 6], [1, 2]], [[2, 4], [3, 4, 5]]]

代码解释 #

• 字典分组：遍历原列表一次，把每个子项按第二个元素（转元组）分类，自动完成去重和收集，时间复杂度是O(n)，比嵌套循环的O(n²)高效得多。
• 格式转换：遍历字典的键值对，把收集到的第一个元素列表和转回的原第二个元素列表组合成组，再加入结果列表。
• 过滤单个元素组：如果不需要只有一个元素的组（比如[[5], [5,6,7]]），只需要加if len(values) > 1的判断即可。

内容的提问来源于stack exchange，提问作者Hal