python - 如何单独处理由 groupby 创建的组中的第一项？

Question

我有一个元组列表 (name, path, ref)已排序，我正在使用 itertools.groupby按 ref 对它们进行分组。

我正在尝试为每组的第一项做一些不同的事情。

代码如下:

from itertools import groupby
from operator import itemgetter

docs = [('foo', '.', 'ref2'), ('bar', '.', 'ref2'), ('baz', '.', 'ref1')]
docs = sorted(docs, key=lambda x: (x[2], x[0]))
for k, g in groupby(documents, itemgetter(2)):
    grouped_docs = list(g)
    print(grouped_docs[0][0], 'came first.')
    for doc in grouped_docs[1:]:
        print(doc[0], 'came after')

这可行，但我必须转换 g到一个列表，然后对其进行切片。

有没有办法获取g的第一项然后在不使用列表的情况下迭代生成器的其余部分？

类似于:

for k, g in groupby(docs, itemgetter(2)):
    first_item = <*magic*>(g)
    print(first_item[0], 'came first.')
    for doc in g:
        print(doc[0], 'came after')

Answer 1

由于它只是一个迭代器，因此使用 next 获取下一个值，然后使用第一个项目(因此它不会出现在 for 循环中。)

for k, g in groupby(documents, itemgetter(2)):
    first_item = next(g)
    print(first_item[0], 'came first.')
    for doc in g:
        print(doc[0], 'came after')