如何解决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 举报,一经查实,本站将立刻删除。