繁体   English   中英

字典为相同的键添加值

[英]dictionary add values for the same keys

我有一个字典列表:

[{'name':'Jay', 'value':'1'},{'name':'roc', 'value':'9'},{'name':'Jay', 'value':'7'},{'name':'roc', 'value':'2'}]

我希望它是:

[{'name':'Jay', 'value':'8'},{'name':'roc', 'value':'11'}]

我试过循环,但我无法找到一个可以做到这一点的例子。 任何提示或想法将不胜感激。

您可以使用defaultdict

lst = [{'name':'Jay', 'value':'1'},{'name':'roc', 'value':'9'},{'name':'Jay', 'value':'7'},{'name':'roc', 'value':'2'}]

1)每个名称的总和值:

from collections import defaultdict    
result = defaultdict(int)

for d in lst:
    result[d['name']] += int(d['value'])

2)将名称 - 值对转换为列表中的字典:

[{'name': name, 'value': value} for name, value in result.items()]
# [{'name': 'roc', 'value': 11}, {'name': 'Jay', 'value': 8}]

或者,如果您希望值为str类型,请注意@Kevin:

[{'name': name, 'value': str(value)} for name, value in result.items()]
​# [{'name': 'roc', 'value': '11'}, {'name': 'Jay', 'value': '8'}]

这是itertools.groupby一个很好的用例。

from itertools import groupby
from operator import itemgetter

orig = [{'name':'Jay', 'value':'1'},
        {'name':'roc', 'value':'9'},
        {'name':'Jay', 'value':'7'},
        {'name':'roc', 'value':'2'}]

get_name = itemgetter('name')
result = [{'name': name, 'value': str(sum(int(d['value']) for d in dicts))}
           for name, dicts in groupby(sorted(orig, key=get_name), key=get_name)]

打破它:

  1. get_name是一个给定字典的函数,返回其“name”键的值。 即, get_name = lambda x: x['name']

  2. sorted返回按"name"键的值排序的字典列表。

  3. groupby返回一个迭代器(name, dicts) ,其中dicts是一个列表(ok,generator),它们共享name作为"name"键的值。 (仅对具有相同键值的连续项进行分组,因此需要在上一步中对列表进行排序。)

  4. 结果是使用给定名称的新词典列表以及所有相关"value"元素的总和

与Psidom的答案类似,但使用collections.Counter ,它是累积整数值的理想选择。

import collections

d =[{'name':'Jay', 'value':'1'},{'name':'roc', 'value':'9'},{'name':'Jay', 'value':'7'},{'name':'roc', 'value':'2'}]

c = collections.Counter()
for sd in d:
    c[sd["name"]] += int(sd["value"])

然后,如果需要,您需要通过转换回字符串来重建dicts。

print([{"name":n,"value":str(v)} for n,v in c.items()])

结果:

[{'name': 'Jay', 'value': '8'}, {'name': 'roc', 'value': '11'}]

为了完整起见,没有collections.defaultdict

data = [{'name': 'Jay', 'value': '1'}, {'name': 'roc', 'value': '9'},
        {'name': 'Jay', 'value': '7'}, {'name': 'roc', 'value': '2'}]

result = {}
# concetrate
for element in data:
    result[element["name"]] = result.get(element["name"], 0) + int(element["value"])
# unpack
result = [{"name": element, "value": result[element]} for element in result]
# optionally, you can loop through result.items()
# you can, also, turn back result[elements] to str if needed

print(result)
# prints: [{'name': 'Jay', 'value': 8}, {'name': 'roc', 'value': 11}]

itertools模块使用groupby解决问题的另一种方法:

from itertools import groupby

a = [{'name':'Jay', 'value':'1'},{'name':'roc', 'value':'9'},{'name':'Jay', 'value':'7'},{'name':'roc', 'value':'2'}]
final = []
for k,v in groupby(sorted(a, key= lambda x: x["name"]), lambda x: x["name"]):
    final.append({"name": k, "value": str(sum(int(j["value"]) for j in list(v)))})

print(final)

输出:

[{'name': 'Jay', 'value': '8'}, {'name': 'roc', 'value': '11'}]

这是使用熊猫的另一种方式

names = [{'name':'Jay', 'value':'1'},{'name':'roc', 'value':'9'},{'name':'Jay', 'value':'7'},
         {'name':'roc', 'value':'2'}]

df = pd.DataFrame(names)
df['value'] = df['value'].astype(int)
group = df.groupby('name')['value'].sum().to_dict()
result = [{'name': name, 'value': value} for name, value in group.items()]

哪个输出:

[{'value': 8, 'name': 'Jay'}, {'value': 11, 'name': 'roc'}]
ld = [{'name':'Jay', 'value':'1'},{'name':'roc', 'value':'9'},{'name':'Jay', 'value':'7'},{'name':'roc', 'value':'2'}]

tempDict = {}
finalList = []


for d in ld:
  name = d['name']
  value = d['value']
  if name not in tempDict:
    tempDict[name] = 0
  tempDict[name] += int(value)

#tempDict => {'Jay': 8, 'roc': 11}

for name,value in tempDict.items():
  finalList.append({'name':name,'value':value})
print(finalList)  

# [{'name': 'Jay', 'value': 8}, {'name': 'roc', 'value': 11}]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM