Python 中的多维字典

Posted 2020-07-11 巫谢

Python中的dict可以实现迅速查找。那么有没有像数组有二维数组一样，有二维的字典呢？比如我需要对两个关键词进行查找的时候。2D dict 可以通过

dict_2d = {‘a‘: {‘a‘: 1, ‘b‘: 3}, ‘b‘: {‘a‘: 6}}

来建立，并通过

dict_2d[‘a‘][‘b‘]

来访问。但是添加一个新的 “key-value”对却比较复杂。对一维字典，可以用

dict_1d = dict()
dict_1d[‘a‘] = 1

或者

dict_1d.update({‘a‘: 1})

来实现。但是”2-D” dictionary 新添一个”key-value”对时，不能简单的用

dict_2d[‘a‘][‘c‘] = 8

的形式。因为二维字典的两层key和value之间会混淆，需要判断第一个key是否已经存在了。添加二维的字典可以通过一个函数来简单实现：

def addtodict2(thedict, key_a, key_b, val): 
    if key_a in adic:
        thedict[key_a].update({key_b: val})
    else:
        thedict.update({key_a:{key_b: val}})

同理，三维数组

def addtodict3(thedict,key_a,key_b,key_c,val):
    if key_a in thedict:
        if key_b in thedict[key_a]:
            thedict[key_a][key_b].update({key_c:val})
        else:
            thedict[key_a].update({key_b:{key_c:val}})
    else:
        thedict.update({[key_a]:{key_b:{key_c:val}})