python - 如何按整数倍对距离较近的项目进行分组

Question

我的目标是按 7 的倍数(即 7,14,21)对下面的列表进行分组

mylist=[1,3,7,8,10,14,15,19,22]

理想结果:[(1,3,7),(8,10,14),(15,19),(22)]

我的尝试:

>>>groups=[]
>>> for x in itertools.groupby(mylist,lambda x: x<=range(7,49,7)):
             groups.append(x)
>>> groups
[(True, <itertools._grouper object at 0x0000000002EBC128>)]

关于如何达到理想结果有什么想法吗？谢谢。

Answer 1

您可以使用:

itertools.groupby(mylist, lambda x: (x - 1) // 7)

您当前的尝试将每个项目与 range 对象进行比较，而不是它生成的值。这是没有意义的，并且是 Python 3.x 中的一个 TypeError。

要将 groupby 对象解压到元组列表:

list(map(lambda g: tuple(g[1]), itertools.groupby(...)))