【Python】读取当前文件夹目录、子目录、子文件

林九生

已于 2023-03-28 01:11:10 修改

阅读量2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： python

于 2023-03-26 18:33:25 首次发布

本文链接：https://blog.csdn.net/linjiuxiansheng/article/details/129782513

Python 专栏收录该内容

40 篇文章

订阅专栏

该代码示例使用Python的os模块遍历目录结构，将文件和子目录以树形结构写入到txt文件中，帮助程序员理解并整理复杂的文件系统。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

需求

整理文件是程序员的一个好习惯，在整理文件的时候，目录和文件太多，但想知道目录结构是怎么样的，再来构思怎么整理文件。

环境

Python：3.6

代码

import os

# 目录遍历方法
def checkDir(path, folder, count):
    # 检查是否是第一级目录,不是的话在目录命名前加斜杠\
    if folder != "":
        folder = folder + '\\'
    # 遍历path目录
    for i, j, k in os.walk(path):
        # 如果该目录下存在非目录文件，则先写入
        if k != []:
            for kk in k:
                print("├{} {}{}".format("─" * count, folder, kk))
                w_file.write("├{} {}{}\n".format("─" * count, folder, kk))
                w_file.flush()
        # 如果存在子目录，则进入下一个目录遍历方法
        if j != []:
            for jj in j:
                print("├{} {}{}".format("─" * count, folder, jj))
                w_file.write("├{} {}{}\n".format("─" * count, folder, jj))
                w_file.flush()
                checkDir("{}\\{}".format(i, jj), jj, count + 1)
        break


if __name__ == '__main__':
    # 新建txt，以上一级文件夹名命名
    w_file = open('{}.txt'.format(os.getcwd().split('\\')[-1]), 'w', encoding='utf-8')
    # 获取当前路径
    filePath = os.getcwd()
    # 写入当前文件路径
    w_file.write("{}".format(filePath))
    print("{}".format(filePath))
    # ”─“计数器
    count = 0
    # 调用目录遍历方法
    checkDir(filePath, "", 0)
    # 写入文件结束标志符
    w_file.write("└ #")
    print("└ #")
    # 关闭文件
    w_file.close()

示例目录：

运行结果

生成的txt: