itertools groupby 对象未正确输出

Question

我试图使用 itertools.groupby 来帮助我按正或负属性对整数列表进行分组，例如：

输入

[1,2,3, -1,-2,-3, 1,2,3, -1,-2,-3] 

将返回

[[1,2,3],[-1,-2,-3],[1,2,3],[-1,-2,-3]]

但是如果我：

import itertools

nums = [1,2,3, -1,-2,-3, 1,2,3, -1,-2,-3]
group_list = list(itertools.groupby(nums, key=lambda x: x>=0))
print(group_list)
for k, v in group_list:
    print(list(v))
>>>
[]
[-3]
[]
[]

但如果我不list() groupby 对象，它会正常工作：

nums = [1,2,3, -1,-2,-3, 1,2,3, -1,-2,-3]
group_list = itertools.groupby(nums, key=lambda x: x>=0)
for k, v in group_list:
    print(list(v))
>>>
[1, 2, 3]
[-1, -2, -3]
[1, 2, 3]
[-1, -2, -3]

我不明白的是，groupby对象是由一对key和_grouper对象组成的迭代器，groupby对象的list()调用不应该消耗_grouper对象吗？

即使它确实消耗了，我是如何从第二个元素中获得[-3] 的？

Answer 1

根据the docs，明确指出推进groupby 对象会使前一个组不可用（实际上是空的）：

返回的组本身就是一个迭代器，它与groupby() 共享底层迭代器。因为源是共享的，所以当groupby() 对象被推进时，之前的组不再可见。因此，如果以后需要该数据，则应将其存储为列表。

基本上，在推进groupby 对象之前，您需要一个将组迭代器转换为lists 的listcomp，而不是直接使用list 构造函数进行list-ifying，替换：

group_list = list(itertools.groupby(nums, key=lambda x: x>=0))

与：

group_list = [(k, list(g)) for k, g in itertools.groupby(nums, key=lambda x: x>=0)]

大多数itertools 模块类型的设计旨在避免隐式存储数据，因为它们旨在用于潜在的巨大输入。如果所有的 groupers 都存储了来自输入的所有数据的副本（并且 groupby 对象必须确保追溯填充它们），它会变得丑陋，并且可能会意外地破坏内存。根据 Python 之禅，通过强制您显式存储值，您不会意外地存储无限量的数据：

显式优于隐式。