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

Question

我想使用布尔键和 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, 3, 5]
True [2, 4, 6]

提前致谢

Answer 1

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>

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

groupby from Python's module itertools does not seem to group its input

python

itertools