Python 遍历目录:使用 os.walk() 函数详解

Python 的 os 模块提供了 os.walk() 函数,用于方便地遍历目录结构。该函数返回一个三元组 (root, dirs, files),分别表示当前遍历到的目录路径、包含的子目录列表和文件列表。

示例代码

import os

# 遍历目录
def traverse_dir(path):
    for root, dirs, files in os.walk(path):
        print('当前目录路径:', root)
        print('包含的子目录列表:', dirs)
        print('包含的文件列表:', files)

if __name__ == '__main__':
    traverse_dir('D:/test') # 遍历 D:/test 目录

代码解析

  • os.walk(path):使用 os.walk() 函数遍历指定的目录 path
  • for root, dirs, files in os.walk(path):循环遍历每个目录,获取 root(目录路径)、dirs(子目录列表)和 files(文件列表)。
  • print():打印每个目录的路径、子目录列表和文件列表。

文件操作

如果需要对每个文件进行操作,可以在循环中添加相应的代码。例如,可以使用 os.path.join() 函数获取文件的完整路径,然后进行文件操作。

import os

# 遍历目录并操作每个文件
def process_files(path):
    for root, dirs, files in os.walk(path):
        for file in files:
            file_path = os.path.join(root, file)
            # 对文件进行操作,例如读取内容、写入内容等
            print('正在处理文件:', file_path)

if __name__ == '__main__':
    process_files('D:/test')

总结

使用 os.walk() 函数可以方便地遍历目录结构,获取目录路径、子目录列表和文件列表。在循环中,可以使用 os.path.join() 函数获取每个文件的完整路径,并进行相应的操作。


原文地址: https://www.cveoy.top/t/topic/ntt5 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录