python - 比较元组列表中的相似值

Question

我有一个元组列表，如下例所示。我怎样才能浏览列表并找到所有相同的缩写，然后将分数加起来。希望能够对所有缩写执行此操作。

loop = [('MD', 0.0), ('MD', 0.0), ('MD', 0.0), ('LA', 0.0625), ('LA', 0.0625), ('CA', -0.5), ('CA', -0.5), ('CA', -0.5))]

Answer 1

您可以使用 itertools.groupby() 来完成

>>> from itertools import groupby
>>> loop = sorted(loop, key = lambda x:x[0])
>>> {k:sum(list(e[1] for e in v)) for k, v in groupby(loop, lambda x:x[0])}
{'MD': 0.0, 'CA': -1.5, 'LA': 0.125}
>>> 

---

要获得平均值，请将上面的代码替换为以下代码:

>>> result = dict()
>>> for k, v in groupby(loop, lambda x:x[0]):
...     temp = list(v)
...     result[k] = sum(i[1] for i in temp)/len(temp)
... 
>>> result
{'MD': 0.0, 'LA': 0.0625, 'CA': -0.5}
>>>