代码段
from functools import reduce
import os
def get_directory_structure(rootdir):
"""
Creates a nested dictionary that represents the folder structure of rootdir
"""
dir = {}
rootdir = rootdir.rstrip(os.sep)
start = rootdir.rfind(os.sep) + 1
for path, dirs, files in os.walk(rootdir):
folders = path[start:].split(os.sep)
subdir = dict.fromkeys(files)
parent = reduce(dict.get, folders[:-1], dir)
print(subdir)
parent[folders[-1]] = subdir
return dir
print(get_directory_structure(r"C:\Users\MZG\Desktop\MzgExercise"))
效果图：
