如何在python中对相同字符的句子进行分组('aabcdd' - > ['aa','b','c','dd'])？

Question

我有一个'aabbababacccssdd'我想要生成的字符串['aa', 'bb', 'a', 'b', 'a', 'b', 'a', 'ccc', 'ss', 'dd']

这是我目前的解决方案:

def get_pats(n):
    n = str(n) # to support integers
    a = len(n)
    p = []
    pat_start = 0
    prev = 0
    for b in range(0, a):
        if n[b] != n[prev]:
            p.append(n[pat_start:b])
            prev = b
            pat_start = b
    p.append(n[pat_start:b+1])
    return p

解决方案工作得很好,但我想知道是否有更优雅/ pythonic的方式来做到这一点？

Answer 1

这就是itertools.groupby为你做的事情:

text = 'aabbababacccssdd'

from itertools import groupby
print [''.join(g) for k, g in groupby(text)]
# # ['aa', 'bb', 'a', 'b', 'a', 'b', 'a', 'ccc', 'ss', 'dd']