python - 如何使用 itertools groupby 进行打印？

Question

我有这个代码:

from itertools import groupby

text = ["a", "a", "b", "b", "c"]

group = groupby(text)

for k, g in group:
    print(k, end= " ")
    print(sum(1 for _ in g), end=" ")

示例我需要什么:

A B C
2 2 1

我的 itertools 只显示如下:

A 2 B 2 C 1

Answer 1

您可以这样做，对 groupby 返回的结果进行后处理，以便轻松获取输出的每一行所需的值:

from itertools import groupby

text = ["a", "a", "b", "b", "c"]

groups = [(k, str(sum(1 for _ in g))) for k, g in groupby(text)]
a, b = zip(*groups)
print(' '.join(a))  # -> a b c
print(' '.join(b))  # -> 2 2 1