使用Python的groupby实现压缩功能

最新推荐文章于 2024-03-29 10:11:30 发布

原创最新推荐文章于 2024-03-29 10:11:30 发布 · 1.8k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #groupby

Python 专栏收录该内容

1 篇文章

订阅专栏

本文介绍如何利用Python itertools库中的groupby函数对数据进行去重与压缩操作，通过实例演示了该函数的功能及应用。

from itertools import groupby

def compress(data):
    return ((name, len(list(group))) for name, group in groupby(data))

def decompress(data):
    return (name * size for name, size in data)

c = compress('get uuuuuuuuuuuuuuuuuuuuuuuup')
print(''.join(decompress(c)))

grouby函数类似Unix命令uniq，对相邻的重复元素进行分组。

groupby(iterable[, keyfunc]) -> create an iterator which returns (key, sub-iterator) grouped by each value of key(value).

迭代sub-iterator得到的就是每一个元素，即key。