如何在不创建子列表的情况下进行游程编码？

Question

感谢您的帮助，我想知道如何创建一个对列表进行行程编码但不创建子列表的函数。原因是每个列表代表二维列表的一行。这样，当我去解码所有内容时，我将能够逐行进行解码，并保留原始列表的行。 如果可能，请修改我附上的代码，因为它通过写入重复值的编号而不是写入仅出现一次的值的编号来进行游程长度编码。

from itertools import groupby
def modified_encode(alist):
        def ctr_ele(el):
            if len(el)>1:
                return [len(el), el[0]]
            else:
                return el[0]
        return [ctr_ele(list(group)) for key, group in groupby(alist)]

我会解释一下。这是使用此处发布的函数的示例：

wall = 'wall'
concrete = 'concrete'
>>> map=[wall, wall, concrete, concrete]
>>> modified_encode(map)
[[2, 'wall'], [2, 'concrete']]

我想要的结果是这样的：

[2, 'wall', 2, 'concrete']

Answer 1

您始终可以返回 list - 即使对于单个项目 [ el[0] ] - 然后您可以使用可以添加列表的事实：

[2,' wall'] + [2,'concrete'] + ['single'] 给出 [2,'wall',2,'concrete','single']

您甚至可以使用总和(..., []) 为了这。

from itertools import groupby

def modified_encode(alist):
    def ctr_ele(el):
        if len(el)>1:
            return [len(el), el[0]]
        else:
            return [ el[0] ]   # <--- return list with single element
    
    result = (ctr_ele(list(group)) for key, group in groupby(alist))
    flatten = sum(result , [])  # <--- add all lists to `[]`

    return flatten

data = ['wall', 'wall', 'concrete', 'concrete', 'single']

modified_encode(data)

结果：

[2, 'wall', 2, 'concrete', 'single']

---

编辑：

或者你可以用不同的方式编写它 - 没有 ctr_ele - 并使用 append()

from itertools import groupby

def modified_encode(alist):
    result = []
    
    for key, group in groupby(alist):
        el = list(group)
        if len(el) > 1:
            result.append(len(el))
        result.append(el[0])
        
    return result

data = ['wall', 'wall', 'concrete', 'concrete', 'single']

modified_encode(data)