Python 文件树目录结构生成脚本

答案:使用os模块遍历目录可生成类似tree命令的树形结构,通过递归调用print_tree函数实现缩进显示层级,支持权限错误处理与排序输出。

基本思路:使用 os 模块遍历目录

要生成一个类似命令行 tree 命令的文件树结构,可以利用 Python 的 os.walk() 或递归调用 os.listdir() 遍历目录。下面是一个简洁实用的脚本示例,能清晰输出目录层级结构。

实现代码:支持缩进显示层级

以下脚本会打印指定路径下的文件和文件夹树形结构:

import os

def print_tree(path, prefix="", is_last=True):

获取当前路径的名称用于显示

basename = os.path.basename(path)
connector = "└── " if is_last else "├── "
print(prefix + connector + basename)

# 如果是文件,不再深入
if os.path.isfile(path):
    return

try:
    entries = os.listdir(path)
except PermissionError:
    print(prefix + ("    " if is_last else "│   ") + "└── [Permission Denied]")
    return

entries.sort()  # 排序保证输出一致
for i, entry in enumerate(entries):
    child_path = os.path.join(path, entry)
    is_last_entry = (i == len(entries) - 1)
    extension = "    " if is_last_entry else "│   "
    print_tree(child_path, prefix + ("    " if is_last else "│   "), is_last_entry)

使用示例

if name == "main": root_path = "." # 可替换为任意路径,如 "/home/user/project" print_tree(root_path)

输出效果示例

运行后输出类似:

.
└── folder1
    ├── file1.py
    └── subfolder
        └── test.txt
└── folder2
│   └── data.json
└── main.py

增强功能建议

  • 添加参数控制是否显示隐藏文件(以 . 开头)
  • 支持输出到文件而非仅打印到终端
  • 过滤只显示特定扩展名的文件
  • 统计总文件/目录数量

这个脚本轻量、无需依赖第三方库,适合集成到项目工具或自动化脚本中。基本上就这些,不复杂但很实用。