将字符串中的字母更改为字母和频率-python的数量[重复]

Question

我正在寻找解决问题的方法。我想制作一个有人输入字符串的程序，然后我将其覆盖为如下内容：

'ZZZZYYYZZ' -> 'Z4Y3Z2'

我愿意接受任何建议。

我做的代码：

def compress(s):
    e={}
    if s.isalpha():
        for i in s:
            if i in e:
                e[i] += 1
            else:
                e[i] = 1
    else:
        return None    

    return ''.join(['{0}{1}'.format(k,v)for k,v in e.items()])

s=input("Write string: ")
compress(s)

这会产生错误的输出

Write string: ZZZZYYYZZ
Y3Z6

Answer 1

将未排序的数据分组为 itertools.groupby 的工作。

>>> from itertools import groupby
>>> 
>>> s = 'ZZZZYYYZZ'
>>> ''.join('{}{}'.format(c, len(list(g))) for c, g in groupby(s))
'Z4Y3Z2'

详细介绍 groupby 在此处生成的内容：

>>> [(c, list(g)) for c, g in groupby(s)]
[('Z', ['Z', 'Z', 'Z', 'Z']), ('Y', ['Y', 'Y', 'Y']), ('Z', ['Z', 'Z'])]

~编辑~

没有中间列表的轻微内存优化：

>>> ''.join('{}{}'.format(c, sum(1 for _ in g)) for c, g in groupby(s))
'Z4Y3Z2'

~编辑2~

我们可以只用 C 代替 C1 吗？

>>> s = 'XYXYXXX'
>>> to_join = []
>>> groups = groupby(s)
>>> 
>>> for char, group in groups:
...:    group_len = sum(1 for _ in  group)
...:    if group_len == 1:
...:        to_join.append(char)
...:    else:
...:        to_join.append('{}{}'.format(char, group_len))
...:        
>>> ''.join(to_join)
'XYXYX3'

Answer 2

这有助于巧妙地使用zip，允许您遍历每个字符和下一个字符：

s = 'ZZZZYYYZZ'
out = ''
count = 1
for a, b in zip(s[:-1], s[1:]):
    print(a, b)
    if a != b:
        out += a + str(count)
        count = 1
    else:
        count += 1

out += s[-1] + str(count)

将out 设为'Z4Y3Z2'。