字典中的值总和与键相关 - Python 2.7

Question

我的字典Dict排列如下。每个键都与一个值列表相关联，其中每个值都是一个元组：

Dict = { 
    'key1': [('Red','Large',30),('Red','Medium',40),('Blue','Small',45)],
    'key2': [('Red','Large',35)],
    'key3': [('Yellow','Large',30),('Red','Medium',30)], 
}

然后我想在给定一个新键的情况下对整数（每个元组的索引 2）求和，在这种情况下为 Color。

生成的新字典应如下所示：

{
    'key1': [('Red', 70), ('Blue', 45)],
    'key2': [('Red', 35)],
    'key3': [('Yellow', 30), ('Red', 30)],
}

我将如何做到这一点？

我在想类似以下的事情，但我知道这在几个方面是错误的。

sum = 0
new_dict = {}
new_key = raw_input("Enter a new key to search on: ")
for k,v in Dict:
  if v[0] == new_key:
    sum = sum + v[2]
    new_dict[k].append(sum)
  else:
    sum = 0
    new_dict[k] = [sum]

Answer 1

使用 dict 理解生成新的输出：

{key: [color, sum(t[2] for t in value if t[0] == color)] for key, value in Dict.iteritems()}

color 是搜索的关键字。

演示：

>>> Dict = {
...     'key1': [('Red','Large',30),('Red','Medium',40),('Blue','Small',45)],
...     'key2': [('Red','Large',35)],
...     'key3': [('Yellow','Large',30),('Red','Medium',30)], 
... }
>>> color = 'Red'
>>> {key: [color, sum(t[2] for t in value if t[0] == color)] for key, value in Dict.iteritems()}
{'key3': ['Red', 30], 'key2': ['Red', 35], 'key1': ['Red', 70]}

要按颜色对所有值求和，请使用Counter() 对值求和：

from collections import defaultdict, Counter

new_dict = {}
for key, values in Dict.iteritems():
    counts = Counter()
    for color, _, count in values:
        counts[color] += count
    new_dict[key] = counts.items()

给出：

>>> new_dict = {}
>>> for key, values in Dict.iteritems():
...     counts = Counter()
...     for color, _, count in values:
...         counts[color] += count
...     new_dict[key] = counts.items()
... 
>>> new_dict
{'key3': [('Red', 30), ('Yellow', 30)], 'key2': [('Red', 35)], 'key1': [('Blue', 45), ('Red', 70)]}