在python中列出目录树结构？

在python中列出目录树结构？

这是一个用于格式化的函数：

import osdef list_files(startpath):    for root, dirs, files in os.walk(startpath):        level = root.replace(startpath, '').count(os.sep)        indent = ' ' * 4 * (level)        print('{}{}/'.format(indent, os.path.basename(root)))        subindent = ' ' * 4 * (level + 1)        for f in files: print('{}{}'.format(subindent, f))