Python——对满足谓词的连续元素进行分组

Question

我想对满足谓词的连续元素进行分组。这种功能的 doctest 看起来像

>>> group([1, 2, 3, 0, 4, 5, 0, 0, 6], lambda x: x != 0)
    [[1, 2, 3], [4, 5], [6]] 
>>> group([1, 2, 3, 0, 4, 5, 0, 0, 6], lambda x: x == 0)
    [[0], [0, 0]]

我已经编写了一个原型(prototype)，其中我使用了 itertool 的 takewhile，但它很难看，因为我一直在 list 和 iter 之间转换。我也不想坚持阅读列表的索引，因为它感觉效率低下。有人能告诉我如何以正确的方式混合和匹配 itertools 吗？

from itertools import takewhile

def group(l, p):
    blocks = []
    while True:
        i = iter(l)
        taken = list(takewhile(p, i))
        l = list(i)
        if len(taken) > 0:
            blocks.append(taken)
        if len(l) == 0:
            return blocks

谢谢!

Answer 1

你可以使用 itertools.groupby :

In [22]: import itertools as IT

In [23]: [list(g) for k, g in IT.groupby(
          [1, 2, 3, 0, 4, 5, 0, 0, 6], lambda x: x != 0) if k]
Out[23]: [[1, 2, 3], [4, 5], [6]]