python 实现二维字典的键值合并等函数
这篇文章主要讲python中关于字典的一些具体操作,讲解的问题都是本人在实际编程中所遇到的问题,读者可以根据自己所遇到的问题具体问题具体分析。
(1) 二维字典的键值合并:
先提供一个应用场景:
假设我有两个二维字典:
room1 = {orderid1:{roomid1:pred1,roomid2:pred2},orderid2:{roomid3:pred3,roomid4:pred4}} room2 = {orderid1:{roomid1:pred5,roomid2:pred6},orderid2:{roomid3:pred7,roomid4:pred8}}
这两个字典的key相同,只有第二维的value不同,我的目的是把这两个字典的value相加,得到的结果如下:
sum_room = {orderid1:{roomid1:pred1+pred5,roomid2:pred2+pred6},orderid2:{roomid3:pred3+pred7,roomid4:pred4+pred8}}
具体的实现代码如下:(为了提高通用性,以下代码创建了两个简单的字典obj3,obj4)
实现二维字典的键值合并,即将同个key下的value相加:
思路:先创建一个空字典all_room,将all_room与obj的value相加并赋给all_room:
from collections import defaultdict all_room = defaultdict(defaultdict) obj3 = {'a':{'aa':1,'aa1':2},'b':{'bb':2},'c':3} obj4 = {'a': {'aa': 5, 'aa1': 6}, 'b': {'bb': 7}, 'c': 8} def sum_value(obj): for key in obj: if type(obj[key]).__name__ == 'dict': if key not in all_room: all_room[key] = {} for subkey in obj[key]: if subkey not in all_room[key]: all_room[key][subkey] = 0 all_room[key][subkey] += obj[key][subkey] else: if key not in all_room: all_room[key] = 0 all_room[key] += obj[key] return all_room
看一下结果:
sum_value(obj3) Out[40]: defaultdict(collections.defaultdict, {'a': {'aa': 1, 'aa1': 2}, 'b': {'bb': 2}, 'c': 3}) sum_value(obj4) Out[41]: defaultdict(collections.defaultdict, {'a': {'aa': 6, 'aa1': 8}, 'b': {'bb': 9}, 'c': 11})
实现了文章开头所述功能。
sum_value(obj)这个函数可以实现对多个二维字典的键值相加,只要创建一个for循环就行了。
(2)往字典的key里添加一个新的value:
同样提供一个应用场景:
dict1 = {'order1': ['room1', 'room1'],'order2': ['room3', 'room3']} dict2 = {'order1': 'room2', 'order2': 'room3'}
我的目的是想把dict2中key里的value添加进dict1中相同的key里,理想结果如下:
dict1 = {'order1': ['room1', 'room1', 'room2'], 'order2': ['room3', 'room3', 'room3']}
下面给出实现代码:
for key in dict1.keys(): a=[] b=[] b = dict1[key] for i in b: ##这一步很关键,目的是把dict1[key]的value值逐个取出存进a,再把dict2[key]的value值存进a里面,这样可以避免dict1[key]出现多个list层 a.append(i) a.append(dict2[key]) ##dict2[key]必须保证只有一个元素 dict1[key] = a
结果如下:
dict1 Out[88]: {'order1': ['room1', 'room1', 'room2'], 'order2': ['room3', 'room3', 'room3']}
下一步,统计key中的value个数:
from collections import Counter for key,item in dict1.items(): count[key] = Counter(item)
结果如下:
count Out[91]: {'order1': Counter({'room1': 2, 'room2': 1}), 'order2': Counter({'room3': 3})}
接着,取出每个外层key中里层value最大的key(比较拗口,即取出每个orderid中roomid的值最大的roomid):
(举个例子,在order1中,room1的值为2,是最大值,所以我把room1赋给order1,即result1 = {'order1':'room1'})
result = {} for key,item in count.items(): result[key] = max(item,key=item.get)
结果如下:
result Out[94]: {'order1': 'room1', 'order2': 'room3'}
(3)字典的排序:sorted()函数
import operator dic = {(1, 2): 3, (1, 3): 4, (4, 5): 14} ##字典的key可以是元组; ds = sorted(dic.items(),key=operator.itemgetter(1),reverse=True) ##如果要升序排列,把reverse=True去掉即可; ds Out[21]: [((4, 5), 14), ((1, 3), 4), ((1, 2), 3)]
(4)字典中不允许出现一个键对应多个值的情况,当出现这种情况时,取最后的键值对:
如:
>>> dict1 = {'foo':789, 'foo': 'xyz'} >>> dict1 结果:{'foo': 'xyz'}
栏 目:Python代码
本文地址:http://www.codeinn.net/misctech/15629.html