给定特定条件时，对嵌套字典列表中的值求和

Question

下面我有一个嵌套字典列表，其中包括给定汽车品牌/型号/年份的汽车销售

    cars = 
    [{'id': 2, 'car': 
                    {'car_make':'Acura', 'car_model':'TL', 'car_year':2005},'total_sales':589}, 
    {'id': 30, 'car': 
                    {'car_make':'Acura', 'car_model':'TL', 'car_year':2004}, 'total_sales':167},
    {'id': 31, 'car': 
                    {'car_make':'Acura', 'car_model':'Integra', 'car_year':2008}, 'total_sales':200},
    {'id': 71, 'car':
                    {'car_make':'BMW', 'car_model':'5 Series', 'car_year':2011},'total_sales':824},
    {'id': 72, 'car':
                    {'car_make':'BMW', 'car_model':'5 Series', 'car_year':2001}, 'total_sales':6}]

我想计算所有年份的总销售额，并返回一本包含汽车品牌、型号和总销售额的总销售额字典

    total_sales = {{'car_make': 'Acura', 'car_model': 'TL', 'total_sales': 756},
                  {'car_make': 'Acura', 'car_model': 'Integra', 'total_sales': 200},
                  {'car_make': 'BMW', 'car_model': '5 Series', 'total_sales': 830}}

下面是我的代码，我在其中迭代嵌套的字典，并添加销售总额

total_sales = {'car_make':{}}

for car in cars:
    if car['car']['car_make'] in total_sales['car_make'] and car['car']['car_model'] in 
    total_sales['car_model']:
        total_sales['total_sales'] = total_sales['total_sales'] + car['total_sales']
else:
    total_sales['car_make'] = car['car']['car_make']
    total_sales['car_model'] = car['car']['car_model']
    total_sales['total_sales'] = car['total_sales']
    print(total_sales)

然而，我得到的是每一个汽车品牌和车型的总销量，而不是最终的总销量

    {'car_make': 'Acura', 'car_model': 'TL', 'total_sales': 589}
    {'car_make': 'Acura', 'car_model': 'TL', 'total_sales': 756}
    {'car_make': 'Acura', 'car_model': 'Integra', 'total_sales': 200}
    {'car_make': 'BMW', 'car_model': '5 Series', 'total_sales': 824}
    {'car_make': 'BMW', 'car_model': '5 Series', 'total_sales': 830}

我是python新手，如果有人能解释我做错了什么，我将不胜感激。我还研究了其他主题，包括迭代嵌套字典，但似乎没有什么适合我的情况

Answer 1

您可以使用collections.defaultdict：

cars = [{'id': 2, 'car': {'car_make': 'Acura', 'car_model': 'TL', 'car_year': 2005}, 'total_sales': 589}, {'id': 30, 'car': {'car_make': 'Acura', 'car_model': 'TL', 'car_year': 2004}, 'total_sales': 167}, {'id': 31, 'car': {'car_make': 'Acura', 'car_model': 'Integra', 'car_year': 2008}, 'total_sales': 200}, {'id': 71, 'car': {'car_make': 'BMW', 'car_model': '5 Series', 'car_year': 2011}, 'total_sales': 824}, {'id': 72, 'car': {'car_make': 'BMW', 'car_model': '5 Series', 'car_year': 2001}, 'total_sales': 6}]
from collections import defaultdict
d = defaultdict(int)
for i in cars:
   d[(i['car']['car_make'], i['car']['car_model'])] += i['total_sales']

r = [{'car_make':a, 'car_model':b, 'total_sales':c} for (a, b), c in d.items()]

输出：

[{'car_make': 'Acura', 'car_model': 'TL', 'total_sales': 756}, 
 {'car_make': 'Acura', 'car_model': 'Integra', 'total_sales': 200}, 
 {'car_make': 'BMW', 'car_model': '5 Series', 'total_sales': 830}]