
本文介绍如何将任意长度的“列表的列表”(每个子列表含若干 (key, value) 元组)扁平化、按键分组,并重组为按相同键聚合的新列表,支持后续按值排序,代码简洁高效。
本文介绍如何将任意长度的“列表的列表”(每个子列表含若干 `(key, value)` 元组)扁平化、按键分组,并重组为按相同键聚合的新列表,支持后续按值排序,代码简洁高效。
在处理结构化嵌套数据时,常需按元组的某个字段(如第一个元素)对分散在多层列表中的数据进行归类与重组。例如,给定如下嵌套结构:
data = [
[('a', 1), ('b', 2), ('c', 4)],
[('c', 2), ('b', 4), ('a', 2)],
[('b', 6), ('c', 7), ('a', 5)]
]目标是将其重排为:
- 所有 'a' 开头的元组组成一个子列表:[('a', 1), ('a', 2), ('a', 5)]
- 所有 'b' 开头的元组组成另一个子列表:[('b', 2), ('b', 4), ('b', 6)]
- 依此类推。
✅ 核心思路:扁平化 → 排序 → 分组 → 提取
该过程可分解为四步,全部使用 Python 内置或标准库工具,无需外部依赖,且天然支持任意长度的输入:
- 扁平化(Flatten):将 list[list[tuple]] 转为单层 list[tuple];
- 按键排序(Sort by key):因 itertools.groupby() 要求分组键连续出现,必须先按元组首项(itemgetter(0))排序;
- 分组(Group by key):利用 groupby(..., key=itemgetter(0)) 将相同键的元组聚合成迭代器;
- 构建结果列表:将每个分组转为 list,得到最终的 list[list[tuple]]。
以下是完整可运行代码:
from itertools import groupby
from operator import itemgetter
data = [
[('a', 1), ('b', 2), ('c', 4)],
[('c', 2), ('b', 4), ('a', 2)],
[('b', 6), ('c', 7), ('a', 5)]
]
# 步骤 1 & 2:扁平化 + 按键排序
flattened_sorted = sorted([t for sublist in data for t in sublist], key=itemgetter(0))
# 步骤 3 & 4:分组并转为列表
grouped_by_key = [list(group) for _, group in groupby(flattened_sorted, key=itemgetter(0))]
print(grouped_by_key)
# 输出:
# [[('a', 1), ('a', 2), ('a', 5)],
# [('b', 2), ('b', 4), ('b', 6)],
# [('c', 2), ('c', 4), ('c', 7)]]? 进阶:按值二次排序(可选)
若还需每个子列表内部按数值(元组第二项)升序排列,只需对结果再映射一次 sorted(..., key=itemgetter(1)):
result_sorted_by_value = [sorted(sublist, key=itemgetter(1)) for sublist in grouped_by_key]
? 提示:由于每个子列表中所有元组首项已相同,也可简写为 [sorted(sublist) for sublist in grouped_by_key],Python 元组默认按索引顺序比较,效果一致。
⚠️ 注意事项
- groupby 不会自动排序,跳过 sorted(..., key=itemgetter(0)) 将导致分组错误(例如 ('a',1), ('c',2), ('a',2) 中两个 'a' 会被拆到不同组);
- 输入中若存在重复键但跨多个子列表,本方案仍能正确聚合;
- 时间复杂度为 O(N log N)(主要由排序决定),空间复杂度 O(N),适用于中等规模数据;
- 若需保持原始出现顺序(而非字典序),可改用 dict.setdefault() 或 defaultdict(list) 实现稳定分组(不排序),但无法直接用 groupby 替代。
此方法简洁、健壮、可扩展,是处理此类嵌套键值聚合任务的标准实践。










