python - 从字典中查找后的同名值求和

Question

一个字符串列表，也就是字符串的那部分，是字典中的一个键。我想通过键从字典中返回值，然后对这些值求和。

d = {"2016_05" : 665,
"2016_04" : 462,
"2015_03" : 568,
"2015_08" : 895}

for a in ['Toyota_2015_03', 'Toyota_2015_04', 'Kia_2016_01', 'Kia_2016_04', 'Kia_2016_05']:
    name, year, month = a.split('_')
    sales = d.get(year + '_' + month)

    if sales is not None:
        print (name + '_' + str(sales))

输出:

Toyota_568
Kia_462
Kia_665

我想通过“Kia”和“Toyota”对数字求和(即某种形式以获得起亚:1127，丰田:568)。什么是继续的好方法？谢谢。

Answer 1

你可以使用字典来保存你的输出:

result = {}
for e in data:
    name, year_month =  e.split('_', 1)
    result[name] = result.get(name, 0) + d.get(year_month, 0)

result

输出:

{'Toyota': 568, 'Kia': 1127}

如果 year_month 不在变量 d 中，您不想使用 0 作为值的键，您可以使用:

result = {}
for e in data:
    name, year_month =  e.split('_', 1)
    sales = d.get(year_month, None)

    if sales is not None:
        result[name] = result.get(name, 0) + sales