计算列表中字符串的连续重复
问题描述
我有一个 Python 字符串列表，
I have a Python list of strings such that,
输入:
li = ['aaa','bbb','aaa','abb','abb','bbb','bbb','bbb','aaa','aaa']
如何生成另一个列表来计算列表中任何字符串的连续重复次数?对于上面的列表，返回列表类似于:
What can I do to generate another list counting the number of consecutive repetitions of any string in the list? For the list above the return list resembles:
预期输出:
li_count = [['aaa',1],['bbb',1]['abb',2],['bbb',3],['aaa',2]]
解决方案
使用 itertools.groupby:
from itertools import groupby
li = ['aaa','bbb','aaa','abb','abb','bbb','bbb','bbb','aaa','aaa']
a = [[i, sum(1 for i in group)] for i, group in groupby(li)]
print(a)
[['aaa', 1], ['bbb', 1], ['aaa', 1], ['abb', 2], ['bbb', 3], ['aaa', 2]]
感谢@user3483203 的改进:
Thank you @user3483203 for improvement:
a = [[i, len([*group])] for i, group in groupby(li)]
相关文章