Question

我有一个目录结构：

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': []}]}

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

Answer 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

递归地添加到Python中的字典

1 个答案: