python - 如何在python中找到运行长度编码

Question

我有一个数组 ar = [2,2,2,1,1,2,2,3,3,3,3]。对于这个数组，我想找到连续相同数字的长度，例如:

 values: 2, 1, 2, 3
lengths: 3, 2, 2, 4

在 R 中，这是通过使用 rle() 函数获得的。 python 中是否有提供所需输出的现有函数？

Answer 1

您可以使用 groupby 执行此操作

from itertools import groupby
ar = [2,2,2,1,1,2,2,3,3,3,3]
print([(k, sum(1 for i in g)) for k,g in groupby(ar)])
# [(2, 3), (1, 2), (2, 2), (3, 4)]