向字典中不存在的键添加值

Question

我有一个包含许多腌制文件的目录，每个文件都包含一本字典。 文件名表示字典的设置。 例如： 20NewsGroup___10___Norm-False___Probs-True___euclidean.pickle 。

我想将这些不同的字典组合成一个大字典。 为此，我编写了以下代码：

PATH = '<SOME PATH>'
all_dicts = os.listdir(PATH)

one_dict = dict()

for i, filename in enumerate(all_dicts):
    infile = open(PATH+filename, 'rb')
    new_dict = pickle.load(infile)
    infile.close()
    splitted = filename.split('___')
    splitted[4] = splitted[4].split('.')[0]
    one_dict[splitted[0]][splitted[1]][splitted[2]][splitted[3]][splitted[4]] = new_dict

但是，当我运行它时，我得到一个KeyError ，因为splitted[0]的键尚不存在。 填充类似于我设想的字典的最佳方法是什么？

Answer 1

你需要在这些领域创造

例子：

from typing import List

PATH = '<SOME PATH>'
all_dicts = os.listdir(PATH)

def check_exist_path(target_dict, paths: List[str]):
     for key in paths:
          if key not in target_dict:
               target_dict[key] = {}

          target_dict = target_dict[key]

one_dict = dict()

for i, filename in enumerate(all_dicts):
    infile = open(PATH+filename, 'rb')
    new_dict = pickle.load(infile)
    infile.close()
    splitted = filename.split('___')
    splitted[4] = splitted[4].split('.')[0]

    for i in range(5):
        check_exist_path(one_dict, [splitted[j] for j in range(i)])

    one_dict[splitted[0]][splitted[1]][splitted[2]][splitted[3]][splitted[4]] = new_dict

Answer 2

你可以使用 try-except 块：

for i, filename in enumerate(all_dicts):
    infile = open(PATH+filename, 'rb')
    new_dict = pickle.load(infile)
    infile.close()
    splitted = filename.split('___')
    splitted[4] = splitted[4].split('.')[0]
    try:
        one_dict[splitted[0]][splitted[1]][splitted[2]][splitted[3]][splitted[4]] = new_dict
    except KeyError:
        one_dict[splitted[0]] = {}
        one_dict[splitted[0]][splitted[1]] = {}
        one_dict[splitted[0]][splitted[1]][splitted[2]] = {}
        one_dict[splitted[0]][splitted[1]][splitted[2]][splitted[3]] = {}
        one_dict[splitted[0]][splitted[1]][splitted[2]][splitted[3]][splitted[4]] = new_dict

此代码尝试访问嵌套的字典键，如果它们不存在，则代码将使用空字典创建它们。