itertools groupby 中没有出现值 - python

问题描述

我正在尝试使用 python - itertools.groupby 对字符串中的值进行分组。我曾尝试使用此代码：

for key,values in itertools.groupby(s):
    print(key,list(values))

我得到这个输出：

a ['a']
b ['b']
a ['a','a']
b ['b','b','b']
c ['c']

这很好。但是当我添加一个 if 条件并将代码更改为这种方式时：

out = ''
for key,values in itertools.groupby(s):
    if len(list(values))==1:
        out+=key
    else:
        out += key
        out += str(len(list(values)))
    print(key,list(values))

我得到这个输出：

a []
b []
a []
b []
c []

我不知道为什么列表是空的

解决方法

values 是一个迭代器，在它用完之前你只能对它调用一次 list()。您应该将该结果保存在一个变量中并重复使用它。

out = ''
for key,values in itertools.groupby(s):
    value_list = list(values)  # values iterator exhausted here,can't use it again
    if len(value_list)==1:
        out+=key
    else:
        out += key
        out += str(len(value_list))
    print(key,value_list)