如何在itertools.groupby中獲取key / group的索引？

Question

當使用itertools的groupby函數時，是否有一種很好的方法來獲取key/group元組的索引？ 我想避免這樣做：

index = 0
for key, group in itertools.groupby(myList, myFunc) :
  ...
  index += 1

例如，可以用枚舉做一些事情嗎？

Answer 1

您需要在括號中放置key和group ：

for index, (key, group) in enumerate(itertools.groupby(myList, myFunc)):

以下是演示：

>>> # lst represents what is returned by itertools.groupby
>>> lst = [(1, 2), (3, 4), (5, 6)]
>>> for i, (j, k) in enumerate(lst):
...     i, j, k
...
(0, 1, 2)
(1, 3, 4)
(2, 5, 6)
>>>