如何在Python中查找列表中的所有组?

4
我有一个像这样的列表
[0,0,1,1,1,0,0,0,1,1,0,0,1,1,1,1] 

我希望将它们分组并找出每个组的长度。因此,结果将如下所示:

[[2,0],[3,1].....[4,1]]
2个回答

8

使用itertools.groupby

>>> import itertools
>>> l =  [0,0,1,1,1,0,0,0,1,1,0,0,1,1,1,1]
>>> [(len(list(g)), k) for k,g in itertools.groupby(l)]
[(2, 0), (3, 1), (3, 0), (2, 1), (2, 0), (4, 1)]

你也可以使用funcy库中的ilen代替len(list(...))以提高速度。 - Suor

2
In [7]: import itertools

In [8]: [[sum(1 for _ in g),v] for v,g in itertools.groupby(l)]
Out[8]: [[2, 0], [3, 1], [3, 0], [2, 1], [2, 0], [4, 1]]

在这里,l代表您的输入列表。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接