如果某个键是sam，则获取所有值的平均值

Question

所以我有一个字典列表：

l = [{'COUNTRY': 'UK', 'CREDITS': '54'}, {'COUNTRY': 'PT', 'CREDITS': '100'}, {'COUNTRY': 'FR', 'CREDITS': '20'}, {'COUNTRY': 'UK', 'CREDITS': '30'}, {'COUNTRY': 'UK', 'CREDITS': '15'}, {'COUNTRY': 'PT', 'CREDITS': '35'}, {'COUNTRY': 'FR', 'CREDITS': '30'}]

我需要得到每个国家的平均学分。输出应该如下所示：

^{pr2}$

有没有简单易行的方法来实现这一点？在

Answer 1

我将首先创建一个defaultdict来收集“COUNTRY”键下整数列表中的值。在

然后我将创建一个dict理解，执行以下意思：

l = [{'COUNTRY': 'UK', 'CREDITS': '54'}, {'COUNTRY': 'PT', 'CREDITS': '100'}, {'COUNTRY': 'FR', 'CREDITS': '20'}, {'COUNTRY': 'UK', 'CREDITS': '30'}, {'COUNTRY': 'UK', 'CREDITS': '15'},
 {'COUNTRY': 'PT', 'CREDITS': '35'}, {'COUNTRY': 'FR', 'CREDITS': '30'}]

import collections

d = collections.defaultdict(list)
for s in l:
    d[s["COUNTRY"]].append(int(s["CREDITS"]))

result = {k:sum(v)/len(v) for k,v in d.items()}

print(result)

结果：

^{pr2}$

请注意，1）您的预期结果是错误的，2）我转换为float，但您可以通过

result = {k:str(sum(v)//len(v)) for k,v in d.items()}

它给出了：

{'PT': '67', 'FR': '25', 'UK': '33'}

Answer 2

使用itertools.groupby()和itertools.tee()函数的替代解决方案：

import itertools

l = [{'COUNTRY': 'UK', 'CREDITS': '54'}, {'COUNTRY': 'PT', 'CREDITS': '100'}, {'COUNTRY': 'FR', 'CREDITS': '20'}, {'COUNTRY': 'UK', 'CREDITS': '30'}, {'COUNTRY': 'UK', 'CREDITS': '15'}, {'COUNTRY': 'PT', 'CREDITS': '35'}, {'COUNTRY': 'FR', 'CREDITS': '30'}]
avgs = {}

for k,g in itertools.groupby(sorted(l, key=lambda x: x['COUNTRY']), key=lambda x: x['COUNTRY']):
    d1,d2 = itertools.tee(g)  # copy `grouper` iterator to deal with "fresh" pointer
    avgs[k] = sum(int(d['CREDITS']) for d in d1)/len(list(d2))

print(avgs)

输出：

^{pr2}$