如何在嵌套字典中重新创建树组织

Question

我有一个问题，我一直在努力解决一段时间。 我需要做的是检查许多文件夹中的大量数据。 为了跟踪所做的事情，我想创建一个 yaml 文件，其中包含我的数据结构的树组织。 因此，目标是创建包含数据的文件夹的嵌套字典。 我制作的脚本正在运行，但它会复制每个文件夹，我不知道如何递归调用 function 来避免这种情况。 这是代码：

def load_tree_structure_as_dictionnary(current_dict):

    for dir_name in current_dict.keys():

        lst_sub_dir = [f.path for f in os.scandir(dir_name) if f.is_dir()]

        if lst_sub_dir == []:

            current_dict[dir_name]['correct_calibration'] = None

        else:
            for sub_dir in lst_sub_dir:

                current_dict[dir_name][sub_dir] = load_tree_structure_as_dictionnary( {sub_dir: {}} )

    return current_dict
init_dict = {data_path : {} }
full_dict = load_tree_structure_as_dictionnary(init_dict)

我知道错误出在递归调用中，但如果没有初始化的字典（因此 {sub_dir: {}} ）我无法创建新的“sub_dir”键需要在语法上进行改进。

Answer 1

将current_dict[dir_name][sub_dir] = load_tree_structure_as_dictionnary( {sub_dir: {}} )更改为current_dict[dir_name].update(load_tree_structure_as_dictionnary( {sub_dir: {}} ))您的代码将不会复制 sub_dir。

def load_tree_structure_as_dictionnary(current_dict):

    for dir_name in current_dict.keys():

        lst_sub_dir = [f.path for f in os.scandir(dir_name) if f.is_dir()]

        if lst_sub_dir == []:

            current_dict[dir_name]['correct_calibration'] = None

        else:
            for sub_dir in lst_sub_dir:

                current_dict[dir_name].update(load_tree_structure_as_dictionnary( {sub_dir: {}} ))

    return current_dict
init_dict = {"venv" : {} }
full_dict = load_tree_structure_as_dictionnary(init_dict)