Itertools 根据相邻项目对项目进行分组

问题描述

我有时会发现自己处于一种情况 - 类似于以下假设 - 但我似乎无法想出令人满意的 Pythonic 解决方案：

假设我们有一个数字列表：

numbers = [1,2,3,5,6,8,9]

我们希望将数字分组到子列表中，这样绝对差正好为 1 的相邻数字将成为同一组的一部分。在这种情况下，期望的结果是：

grouped_numbers = [[1,3],[5,6],[8,9]]

从表面上看，itertools.groupby 似乎是完成这项工作的工具。然而，这种分组操作不同于 itertools.groupby 的平均、琐碎的应用，因为 key（返回不同“类别对象”的可调用对象，对应于给定项目的不同组应该属于）必须在 itertools.groupby 的调用之间保留某种状态，因为“当前”组是终止还是继续取决于前一项。

实现此目的的一种方法是使用持久默认参数。这个解决方案有效，但感觉真的很笨拙：

from itertools import groupby

def key(number,prev_number=[None],prev_key=[object()]):
    if prev_number[0] is None or abs(number - prev_number[0]) != 1:
        prev_key[0] = object()
    prev_number[0] = number
    return prev_key[0]

print([list(group) for _,group in groupby(numbers,key=key)])

当然，必须存在更多 Python 解决方案，我的问题是：它们是什么？我更喜欢使用 itertools 配方或标准库中可用的任何解决方案。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）