Skip to main content
 首页 » 编程设计

python之使用 groupby 计算列表中的字符串开关

2025年05月04日22rubylouvre

我头疼,字典是这样的:

{a: ['+','+','-','-','+','-','-','+'], 
b: ['+','+','+','-','-','+','+','+','-'], 
c: ['-','-','-','+','+','+']} 

我想知道值字符串更改了多少次,如下所示:

a = 4 
b = 3 
c = 1 

我尝试使用 itertools 中的 groupby,方法是:

for k, v in mydict.iteritems(): 
    print k + ' ' + str([len(list(g[1])) for g in groupby(list(v)) if g[0] =='+']) 

但我只得到一个包含两个值的列表(两个字符串开关).. 我尝试更改“+”和“-”字符但没有成功,有什么建议吗?

请您参考如下方法:

您可以做的是使用字典推导式构建字典,并计算字典中每个条目中的值更改了多少次:

d = {'a': ['+','+','-','-','+','-','-','+'], 'b': ['+','+','+','-','-','+','+','+','-'],  
     'c': ['-','-','-','+','+','+']} 
 
s = {k : sum(1 for i,j in zip(d[k][1:], d[k][:-1]) if i != j) for k in d} 
# {'a': 4, 'b': 3, 'c': 1} 

或者如果你喜欢 itertools.groupby :

{k : sum(1 for i in groupby(d[k]))-1 for k in d} 
# {'a': 4, 'b': 3, 'c': 1} 

详情

为了检测每个列表中元素的变化,您可以将其元素与列表的滞后版本进行比较,这可以通过压缩列表的两个切片来完成,zip(d[k] [1:], d[k][:-1]).

例如,对于键 a:

list(zip(d['a'][1:], d['a'][:-1])) 
# [('+', '+'), ('-', '+'), ('-', '-'), ('+', '-'), ('-', '+'), ('-', '-'), ('+', '-')] 

现在只需使用生成器表达式并在每次元组中的值不同时添加 1:

sum(1 for i,j in zip(d['a'][1:], d['a'][:-1]) if i != j) 
# 4