Python 模块 itertools 中的 groupby 似乎没有对其输入进行分组

问题描述

我想使用布尔键和 itertools 模块中的 groupby 函数设置偶数字典。

代码似乎可以很好地按偶数和非偶数排序，但所有 True 和 False 键都应该在字典中组合在一起，而它们不是。

代码：

# Set a dictionary of even numbers
from itertools import groupby
ls = [1,2,3,4,5,6]

def evens(arg):
  return arg%2==0

group_obj = groupby(ls,key=evens)

for key,value in group_obj:
  print(key,list(value))

输出：

False [1]
True [2]
False [3]
True [4]
False [5]
True [6]

预期输出：

False [1,5]
True [2,6]

提前谢谢您

解决方法

itertools.groupby 不对数据进行排序。您需要自己对其进行排序，通常使用相同的 key。所以这正是你想要的：

group_obj = groupby(sorted(ls,key=evens),key=evens)

输出：

C:\tmp>python x.py
False [1,3,5]
True [2,4,6]

C:\tmp>