Python 实现 itertools.groupby

zltzlt

Python 实现 itertools.groupby

def groupby(iterable, key=None):
    key, cache, elem, flag = (lambda x: x) if key is None else key, [], None, False
    for i in iterable:
        if cache and elem != key(i):
            yield elem, cache
            cache = []
        elem, flag = key(i), True
        cache.append(i)
    if flag:
        yield elem, cache

如果代码有问题，欢迎在评论区指出！