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

Question

我头疼，字典是这样的：

{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] =='+'])

但我只得到一个包含两个值的列表（两个字符串开关）.. 我已尝试更改 '+' 和 '-' 字符但没有成功，有什么建议吗？

Answer 1

您可以做的事情是使用字典理解构建字典并计算字典中每个条目中的值更改的次数：

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

Answer 2

您不需要过滤。只计算groupby：

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

for k,v in mydict.items():
  print(k + ' ' + str(len(list(itertools.groupby(list(v))))-1))