以干净的编程方式创建重度嵌套的python词典

Question

我正在使用一系列函数来填充重度嵌套的字典。 我想知道是否有更简洁的方法来执行此操作，而不仅仅是长分配字符串，如下例所示。

outputdict = {}
outputdict['x']={}
outputdict['x']['y']={}
outputdict['x']['y']['total_patients']=len(example_dict.keys())
outputdict['x']['y']['z']={}
for variable1 in variable1s:
    outputdict['x']['y']['z'][str(variable1)]={}
    outputdict['x']['y']['z'][str(variable1)]['total_patients']=function_1(example_dict, variable1).count()
    for popn in ['total','male','female']:
        outputdict['x']['y']['z'][str(variable1)][popn]={}
        for age_bucket in np.linspace(40,60,5):
            age_str = str(age_bucket)+'_'+str(age_bucket+5)
            outputdict['x']['y']['z'][str(variable1)][popn][age_str]={}
            outputdict['x']['y']['z'][str(variable1)][popn]["total"]={}
            for res in restypes:
                if popn == 'total':
                    codelist, ncodes = function_2(function_1(example_dict, variable1), res, age_bucket)
                else:
                    codelist, ncodes = function_2_gender(function_1(example_dict, variable1), res, age_bucket, popn)
                outputdict['x']['y']['z'][str(variable1)][popn][age_str][res]={}
                outputdict['x']['y']['z'][str(variable1)][popn][age_str][res]['total_codes']=ncodes
                outputdict['x']['y']['z'][str(variable1)][popn][age_str][res]['top_codes']=[]
                for item in codelist:
                    disp = {"code": item[0][:2], "value":item[0][2], "count":item[1]}

                    outputdict['x']['y']['z'][str(variable1)][popn][age_str][res]['top_codes'].append(disp)


                codelist, ncodes = list_top_codes(function_1(example_dict, variable1), res)
                outputdict['x']['y']['z'][str(variable1)][popn]["total"][res]={}
                outputdict['x']['y']['z'][str(variable1)][popn]["total"][res]['top_codes']=[]
                for item in codelist:
                    disp = {"code": item[0][:2], "value":item[0][2], "count":item[1]}
                    outputdict['x']['y']['z'][str(variable1)][popn]["total"][res]['top_codes'].append(disp)
outputdict

Answer 1

你可以使用自动激活与defaultdict 。 这将允许您跳过空dicts的创建，因为当取消引用未定义的键时它们将自动创建：

from collections import defaultdict

dd = lambda: defaultdict(dd)

d = dd()
d['foo']['bar']['foobar'] = 1

所以你的代码看起来如下：

outputdict = dd()
outputdict['x']['y']['total_patients']=len(example_dict.keys())

for variable1 in variable1s:
    outputdict['x']['y']['z'][str(variable1)]['total_patients']=function_1(example_dict, variable1).count()

另一种可能的改进是将嵌套字典存储到变量中，这样您就不必在任何地方键入完整路径：

for variable1 in variable1s:
    nested = dd()
    outputdict['x']['y']['z'][str(variable1)]=nested
    nested['total_patients']=function_1(example_dict, variable1).count()

Answer 2

您可以通过使用dict的defaultdict来避免字典初始化。 您可以根据需要链接这些以创建嵌套字典层次结构。 例如，这是一个2级和3级层次结构字典。

two_level = defaultdict(lambda: defaultdict(dict))
three_level = defaultdict(lambda: defaultdict(lambda: defaultdict(dict)))

你的字典现在如下： two_level[1][2]和three_level[1][2][3]并且将是空的dicts {} 。

所以在你的情况下，你似乎有一个4级嵌套，所以我可能初始化outputdict为：

output_dict = defaultdict(lambda: defaultdict(lambda: defaultdict(lambda: defaultdict(dict))))

想不出你在这里可以做的任何其他事情 - 如果可能的话，我建议你简化这个嵌套结构。