python獲取某路徑下的目錄結構

查看目錄下所有的子目錄和子文件

## 獲得某個路徑下所有文件夾與子文件的樹形結構
import json, os

def list_dir(path, res):
    for i in os.listdir(path):
        temp_dir = os.path.join(path, i)
        if os.path.isdir(temp_dir):
            temp = {"dirname": temp_dir, 'child_dirs': [], 'files': []}
            res['child_dirs'].append(list_dir(temp_dir, temp))
        else:
            res['files'].append(i)
    return res

def get_config_dirs():
    res = {'dirname': 'root', 'child_dirs': [], 'files': []}  # 當前路徑認爲是root根目錄，向其子文件夾與子文件填充
    return list_dir(r'data/aclImdb/train', res)   # 輸入路徑

if __name__ == '__main__':  
    print(json.dumps(get_config_dirs()))