Python os.walk 在遍历目录中的实际应用步骤("Python os.walk 实战：详解遍历目录的步骤与应用")

原创

ithorizon 6个月前 (10-20) 阅读数 16 #后端开发

Python os.walk 实战：详解遍历目录的步骤与应用

一、引言

在Python中，`os`模块提供了许多用于文件和目录操作的函数，其中`os.walk`是一个非常有力的工具，用于遍历指定目录及其子目录下的所有文件和文件夹。本文将详细介绍`os.walk`的使用方法，并通过实际应用场景来展示其有力的功能。

二、os.walk 基础知识

`os.walk`函数返回一个三元组，其中包含当前路径、该路径下的目录列表以及文件列表。其基本用法如下：


import os
for root, dirs, files in os.walk(path):
    # root 描述当前遍历的目录路径
    # dirs 描述当前目录下的所有子目录
    # files 描述当前目录下的所有文件
    pass

三、os.walk 实战步骤

下面将通过一个实际案例来展示怎样使用`os.walk`遍历目录，并获取目录下所有文件和文件夹的信息。

步骤 1：导入os模块


import os

步骤 2：定义要遍历的目录路径


path = '/path/to/directory'

步骤 3：使用os.walk遍历目录


for root, dirs, files in os.walk(path):
    print(f'当前目录: {root}')
    print('子目录列表:', dirs)
    print('文件列表:', files)
    print('-' * 50)

四、os.walk 实际应用场景

下面将介绍几个`os.walk`在实际开发中的应用场景。

场景 1：统计目录下文件的总大小

我们可以通过遍历目录，并累加每个文件的大小来计算总大小。


total_size = 0
for root, dirs, files in os.walk(path):
    for file in files:
        file_path = os.path.join(root, file)
        total_size += os.path.getsize(file_path)
print(f'目录 {path} 下所有文件的总大小为: {total_size} 字节')

使用`os.walk`可以方便地复制整个目录结构。


import shutil
def copy_directory(src, dst):
    if not os.path.exists(dst):
        os.makedirs(dst)
    for root, dirs, files in os.walk(src):
        for file in files:
            src_file_path = os.path.join(root, file)
            dst_file_path = os.path.join(dst, os.path.relpath(src_file_path, src))
            os.makedirs(os.path.dirname(dst_file_path), exist_ok=True)
            shutil.copy2(src_file_path, dst_file_path)
copy_directory('/path/to/source', '/path/to/destination')

场景 3：查找特定类型的文件

通过遍历目录，我们可以找到所有指定扩展名的文件。


def find_files_with_extension(path, extension):
    result_files = []
    for root, dirs, files in os.walk(path):
        for file in files:
            if file.endswith(extension):
                result_files.append(os.path.join(root, file))
    return result_files
python_files = find_files_with_extension(path, '.py')
print('找到的Python文件列表:', python_files)