python - 字典中特定键的总和值

Question

我有一个看起来像这样的字典列表:

source_dict = [{'ppl': 10, 'items': 15, 'airport': 'lax', 'city': 'Los Angeles', 'timestamp': 1, 'region': 'North America', 'country': 'United States'},
{'ppl': 20, 'items': 32, 'airport': 'JFK', 'city': 'New York', 'timestamp': 2, 'region': 'North America', 'country': 'United States'},
{'ppl': 50, 'items': 20, 'airport': 'ABC', 'city': 'London', 'timestamp': 1, 'region': 'Europe', 'country': 'United Kingdom'}... ]

#Gets the list of countries in the dict
countries  = list(set(stats['country'] for stats  in source_dict))

我知道我可以使用一个集合:

    counter = collections.Counter()

    for d in source_dict:
        counter.update(d)

但是，想要按国家/地区分组并仅获取某些键的总数而不是所有键的总数。

所以结果应该是

{'Country': 'United States', 'p95': 30, 'items':37},
{'Country': 'England', 'ppl': 50, 'items':20},...

我不确定如何将多个键合并到一个计数器中。产生那个结果

Answer 1

这是一种使用 collections.defaultdict 和 collections.Counter 的方法。

例如:

from collections import defaultdict, Counter

source_dict = [{'ppl': 10, 'items': 15, 'airport': 'lax', 'city': 'Los Angeles', 'timestamp': 1, 'region': 'North America', 'country': 'United States'},
{'ppl': 20, 'items': 32, 'airport': 'JFK', 'city': 'New York', 'timestamp': 2, 'region': 'North America', 'country': 'United States'},
{'ppl': 50, 'items': 20, 'airport': 'ABC', 'city': 'London', 'timestamp': 1, 'region': 'Europe', 'country': 'United Kingdom'} ]

result = defaultdict(Counter)
for stats in source_dict:
    result[stats['country']].update(Counter({'ppl': stats['ppl'], "items": stats['items']}))

#result = [{'Country': k, **v} for k, v in result.items()]  #Required output
print(result) 

输出:

defaultdict(<class 'collections.Counter'>,
            {'United Kingdom': Counter({'ppl': 50, 'items': 20}),
             'United States': Counter({'items': 47, 'ppl': 30})})