python限制重复字母

Question

将重复字母限制为 1 和 2 的最佳方法是什么，例如:
appppppppple => 苹果和苹果
bbbbbeeeeeer => ber, beer, bber, bbeer

现在，我有这个:

a = "hellllllllllooooooooooooo"
    match = re.search('(.)\\1+', a)

    if match:
        print 'found'
        print re.sub('(.)\\1+', '\\1', a)
        print re.sub('(.)\\1+', '\\1\\1', a)
    else:
        print 'not found'

但它只返回:

helo
helloo

我怎样才能让它按照我想要的方式工作？

Answer 1

不要为此使用 RE。 RE 适用于搜索、匹配和转换，但不适用于生成 字符串。

我们可以将字符串视为向量；每个字母都是一个维度，重复次数是组件沿该维度的长度。给定一个向量 V，您需要与 V 具有相同维度的所有可能向量，这样如果 V 的对应分量为 1，则每个分量的值为 1，否则为 1 或 2。基于此，这里有一个功能可以满足您的需求。

def doppelstring(s):
    letter_groups = ((val, list(group)) for val, group in itertools.groupby(s))
    max_vector = ((val, min(len(group), 2)) for val, group in letter_groups)
    vector_components = ([dim * (l + 1) for l in range(maxlen)] for dim, maxlen in max_vector)
    return [''.join(letters) for letters in itertools.product(*vector_components)]

这是一个使用切片的更紧凑的版本。它的可读性可能有点差，但至少它保持在 78 个字符的限制内:

def doppelstring(s):
    max_vs = (''.join(itertools.islice(g, 2)) for k, g in itertools.groupby(s))
    components = ([s[:l + 1] for l in range(len(s))] for s in max_vs)
    return [''.join(letters) for letters in itertools.product(*components)]