python - 替换 Python 列表中的重复元素？

Question

例如，如果我有如下列表:

[3, 3, 3, 3, 3, 3, 100, 1, 1, 1, 1, 1, 1, 200, 3, 3, 3, 100, 1, 1, 1]

如何删除重复的元素并表示相同的元素后跟重复的次数？ 示例输出:

[3, 6, 100, 1, 6, 200, 3, 3, 100, 1, 3]

其中 3 重复 6 次... 1重复6次……以此类推

Answer 1

您可以使用 itertools.groupby这里有一个生成器函数:

>>> from itertools import groupby                                              
>>> lst = [3, 3, 3, 3, 3, 3, 100, 1, 1, 1, 1, 1, 1, 200, 3, 3, 3, 100, 1, 1, 1]
>>> def solve(seq):
    for k, g in groupby(seq):
        length = sum(1 for _ in g)
        if length > 1:
            yield k
            yield length
        else:
            yield  k
...             
>>> list(solve(lst))
[3, 6, 100, 1, 6, 200, 3, 3, 100, 1, 3]