python - 将列表压缩为带重复的字符串

Question

逻辑:将一个列表压缩成一个字符串，其中在列表中重复两次的字母在结果字符串中重复指定的次数。

例子:

• ['a','a',2,'b','c','c',3], --> aabccc
• ['a,','a',2] --> aa
• ['c','c',3] --> ccc.

我的代码是:

def run_length_decoder(in_list):
list=[]
for i in range(0,len(in_list)-1):
    if in_list[i]==in_list[i+1]:
        for x in range((i+2)-1):
            list.append(in_list[i])
    else:
        list.append(in_list[i])
list.append(in_list[-1])
word = ""
for letter in list:
    word += str(letter)
return word

这行不通...因为使用 ['a', 'a', 2, 'b', 'b', 2, 'a', 'b', 'a 运行上面的代码'] 返回 'aaa2bbbbbb2aba'，但应该返回 'aabbaba'。

Answer 1

看起来很复杂。对于初学者，您应该更正缩进。此外，由于数字计数器是可选的，因此您需要添加逻辑来识别整数与字符串。最后，由于字母数不需要与以下计数器同步，因此您需要明确地为每个计数器附加一个字母。

您可以通过 itertools.groupby 和列表推导来执行此操作。

from itertools import groupby, zip_longest

L = ['a', 'a', 2, 'b', 'b', 2, 'a', 'b', 'a']

L2 = [i for i, _ in groupby(L)]

res = ''.join([i*(j if isinstance(j, int) else 1)
               for i, j in zip_longest(L2, L2[1:])
               if isinstance(i, str)])

print(res)

aabbaba