python - 递归只打印一个列表

Question

在下面的代码中，我返回给定字符串中连续数字数量的整数值。

def consecutive_length(S):
    if S == '':
        return 0
    if len(S) == 1:
        return 1
    if S[0] == S[1]:
        return 1 + consecutive_length(S[1:])
    return 1

def compress(S):
    if S == '':
        return 0
    cons_length = consecutive_length(S)
    return [cons_length] + [compress(S[cons_length:])]

当我运行此打印语句时，将返回以下内容:

>>> print (compress('1111000000001111000111111111111111'))
[4, [8, [4, [3, [15, 0]]]]]

我真正希望返回以下内容:

>>> print (compress('1111000000001111000111111111111111'))
[4, 8, 4, 3, 15]

Answer 1

您的方法的替代方法是使用itertools.groupby():

from itertools import groupby

s = '1111000000001111000111111111111111'
answer = [len([digit for digit in group[1]]) for group in groupby(s)]
print(answer)

输出

[4, 8, 4, 3, 15]