微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

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

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?