根据定义的值“拆分”RLE（groupby）的输出（拆分 RLE 的“字符”）

Question

考虑“字符串”（将其视为数字数组）

0 0 1 8 8 8 1 0

RLE（“groupby”）是：

[(0,2), (1, 1), (8,3), (1, 1), (0, 1)]

然后我们用前面元素的游程总和来丰富上述 RLE。

因此，上述内容的丰富版本变为：

[(0, (0,2)), (0+2, (1, 1)), (0+2+1, (8,3)), (0+1+2+3, (1, 1)), (0+1+2+3+1, (0, 1))]

“字符串”在 1 处拆分：

0 0 , 8 8 8 , 0

RLE 在 1 上拆分

[(0,2)] , [(8,3)] , [(0, 1)]

“字符串”在 8 上拆分：

0 0 1 , , , 1 0

RLE 在 8 上拆分

[(0,2), (1, 1)] , , , [(1, 1), (0, 1)]

注意：在我的示例中，我引用了“RLE 在 Z 上拆分”列表，但并未对其进行丰富。事实并非如此。我把它们排除在外以减少混乱。例如，“RLE split on 1”实际上应该被视为：

[(0, (0,2))] , [(0+2+1, (8,3))] , [(0+1+2+3+1, (0, 1)]

我怎样才能在 Z 上实现这种“RLE 拆分”（= 1, 8; 在这种情况下）

可以省略空数组（拆分后）。

也许是一个聪明的列表组合？ （使用嵌套了 append 的 for 循环似乎更容易解决）

Answer 1

只是为了说明如何，我强烈建议你不要使用这个

“优雅”的丑陋方式：

>>> data
[0, 0, 1, 8, 8, 8, 4, 4, 1, 0]
>>> def fromDataToSplitRLE(dat,n):
    RLE=[(k,len(tuple(g))) for k,g in itertools.groupby(dat)]
    tmp=tuple(zip(*RLE))
    return [list(g) for k,g in itertools.groupby((zip((sum(tmp[1][:i]) for i in range(len(tmp[1]))) ,(zip(*tmp)))),lambda x:x[1][0]!=n) if k]

>>> fromDataToSplitRLE(data,1)
[[(0, (0, 2))], [(3, (8, 3)), (6, (4, 2))], [(9, (0, 1))]]

Answer 2

import itertools

def get_rle(list_of_digits, split_on=None):
    count = 0
    rle = []
    active_group = []
    rle_app = rle.append
    for item, group in itertools.groupby(list_of_digits):
        L = len(list(group))
        if item == split_on:
            rle_app(active_group)
            active_group = []
        else:
            active_group.append((count, (item, L)))
        count += L

    rle_app(active_group)
    return rle

list_of_digits = map(int, '0 0 1 8 8 8 1 0'.split())
print get_rle(list_of_digits)
print get_rle(list_of_digits, 8)
print get_rle(list_of_digits, 1)

aaron@aaron-laptop:~/code/tmp$ python rle.py
[[(0, (0, 2)), (2, (1, 1)), (3, (8, 3)), (6, (1, 1)), (7, (0, 1))]]
[[(0, (0, 2)), (2, (1, 1))], [(6, (1, 1)), (7, (0, 1))]]
[[(0, (0, 2))], [(3, (8, 3))], [(7, (0, 1))]]