在Python中递归添加到字典答案

【问题标题】：Recursively adding to dictionary in Python在Python中递归添加到字典
【发布时间】：2014-12-07 01:11:16
【问题描述】：

我有一个目录结构：

test
    test2 
        test3

我正在尝试制作一个看起来像这样的字典：

{'name': 'test', 'sub': [{'name': 'test2', 'sub': [{'name': 'test3', 'sub': []}]}, ]}

为此，我使用以下递归函数：

def dir_struct(rootdir):
    a = {}
    rootdir = rootdir.rstrip(os.sep)
    a["name"] = rootdir
    a["sub"] = []
    for path, dirs, files in os.walk(rootdir):
        for dir in dirs:
            b = dir_struct(dir)
            a["sub"].append(b)
    return a

但有了它我得到了

{'name': 'test', 'sub': [{'name': 'test2', 'sub': []}, {'name': 'test3', 'sub': []}]}

有没有办法解决这个问题？

【问题讨论】：

标签： python recursion dictionary directory

【解决方案1】：

问题是os.walk已经是一个递归函数，它遍历整个目录结构。

一种解决方案是使用os.listdir 而不是os.walk。
这就是你想要的：

def dir_struct(rootdir):
    a = {}
    rootdir = rootdir.rstrip(os.sep)
    a["name"] = rootdir.split(os.sep)[-1]
    a["sub"] = []
    for dir in os.listdir(rootdir):
        b = dir_struct(rootdir + os.sep + dir)
        a["sub"].append(b)
    return a

【讨论】：

谢谢你，弗里迪！像魅力一样工作！