对itertools.groupby的结果使用zip会意外地给出空列表

itertools.groupby 是 Python 标准库中的一个函数，它用于对连续的数据进行分组。groupby 函数根据输入迭代器中的元素的某个键函数进行分组，返回一个迭代器，该迭代器产生一系列的（key, group）对，其中 key 是分组的键，group 是一个迭代器，包含了所有具有相同键的元素。

zip 函数用于将多个可迭代对象打包成一个元组列表，每个元组包含来自每个可迭代对象的元素。

当你尝试将 itertools.groupby 的结果与 zip 结合使用时，可能会遇到得到空列表的情况。这通常是因为 zip 函数在处理较短的可迭代对象时会提前终止，而 groupby 生成的分组可能比 zip 预期的要长。

例如，假设你有以下代码：

import itertools

data = [('a', 1), ('a', 2), ('b', 3), ('c', 4)]
key_func = lambda x: x[0]

grouped = itertools.groupby(data, key_func)
zipped = zip(grouped)

print(list(zipped))

在这个例子中，groupby 会根据第一个元素对数据进行分组，得到以下分组：

('a', [('a', 1), ('a', 2)])
('b', [('b', 3)])
('c', [('c', 4)])

然而，当你尝试将这些分组与 zip 结合时，zip 会尝试将这些分组打包成元组，但由于 zip 无法处理不同长度的可迭代对象，它会提前终止，导致得到一个空列表。

要解决这个问题，你可以先将 groupby 的结果转换为列表，然后再进行 zip 操作：

import itertools

data = [('a', 1), ('a', 2), ('b', 3), ('c', 4)]
key_func = lambda x: x[0]

grouped = list(itertools.groupby(data, key_func))
zipped = zip(*grouped)

print(list(zipped))

在这个修改后的例子中，grouped 是一个包含分组元组的列表，zip(*grouped) 会将这些分组元组的元素分别打包成元组，得到以下结果：

(('a', 'a'), (1, 2))
(('b',), (3,))
(('c',), (4,))

这样，你就可以避免得到空列表的问题。

参考链接：

• itertools.groupby
• zip