Python如何实现Counter字典连续键对应值的Pythonic分组

核心逻辑前置说明 #

实现前必须先做一步预处理：将字典的键按数值升序排序——毕竟Counter作为字典子类，键的存储顺序仅匹配插入顺序，不保证按数值大小排列，跳过这步会直接导致分组逻辑失效。
连续序列的判定可以复用itertools.groupby的分组能力，核心判定规则是：排序后的键值序列中，连续数值的键与其自身位置索引的差值为固定值。比如排序后键序列是[0,1,4,5,6]，对应索引位置是[0,1,2,3,4]，计算每个键和索引的差值：

• 0-0=0、1-1=0 → 差值一致，属于同一连续段
• 4-2=2、5-3=2、6-4=2 → 差值一致，属于同一连续段

可直接运行的实现代码 #

import itertools

my_dict = {0: 1, 1: 1, 4: 1, 5: 2, 6: 1}
# 按键值升序排序键值对
sorted_kv = sorted(my_dict.items())
# 按连续规则分组，提取每组的计数值c
result = [
    [kv[1] for kv in group]
    for _, group in itertools.groupby(
        sorted_kv,
        key=lambda item: item[0] - next(itertools.count())
    )
]

针对示例输入，代码运行后得到的result值为[[1, 1], [1, 2, 1]]，和预期完全匹配。

后续筛选逻辑补充 #

拿到分组结果后，可直接通过列表推导筛选出连续长度≥3的块：

# 筛选连续长度满足要求的块
valid_long_blocks = [block for block in result if len(block) >= 3]

常见踩坑提醒 #

很多人尝试用itertools实现时容易出错的两个点：

• 没有提前对字典键做数值排序，依赖字典的原生插入顺序做分组，遇到键乱序的场景就会得到错误结果
• 给groupby传入的分组key直接使用键本身，没有做连续段的差值计算，导致非连续的同值键被错误分到同一组

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