运行长度编码Python

时间:2017-10-04 18:43:25

标签: python

问题是:编写一个函数,它接受一个字符列表作为参数,并返回一个反映该列表的行程编码的列表。返回列表中的元素遵循以下模式:偶数索引包含字符,奇数索引包含字符连续重复的次数。

当我在函数中键入一个列表时,它不会遍历列表的整个长度。它适用于较短长度的列表,但不适用于较长的列表。不确定如何修复它。

def runLengthEncoding(myList):
    aList=[]
    count=1
    for i in range(1,len(myList)):
        if myList[i] == myList[i-1]:
            count=count+1
        else:
            aList.append(myList[i-1])
            aList.append(count)
            count=1
            if i == (len(myList)-1):
                aList.append(myList[i])
                aList.append(count)
    return aList

1 个答案:

答案 0 :(得分:1)

当最后一个项目重复或者列表长度为1时,您的代码失败:

>>> runLengthEncoding("fooobar")
['f', 1, 'o', 3, 'b', 1, 'a', 1, 'r', 1]
>>> runLengthEncoding("fooobarr")
['f', 1, 'o', 3, 'b', 1, 'a', 1]
>>> runLengthEncoding("a")
[]

您的解决方案的修复方法是删除以if i == (len(myList)-1):开头的三行,并将其替换为

if myList:
    aList.append(myList[-1])
    aList.append(count)

将该块移出循环。

然而,实现行程编码的最简单方法可能是使用itertools模块:

from itertools import chain, groupby

def run_length(iterable):
    return list(chain.from_iterable(
        (val, len([*thing]))
        for val, thing in groupby(iterable)
    ))
相关问题