使用Python os.walk遍历文件夹

Python中的os.walk()函数是一个伟大的工具，可以用来迭代遍历指定目录下的所有文件夹及其子文件夹。本文将提供对Python os.walk函数的详细介绍和代码示例，帮助读者理解其实现的原理。

一、os.walk()函数介绍

os模块提供多个有关文件和目录操作的函数，其中os.walk()函数与os.listdir()函数不同，os.walk()可以返回指定目录及其所有子目录的文件名和目录名，os.listdir()只返回指定目录下的文件名。

os.walk()参数介绍：

os.walk()的第一个参数是欲遍历的目录的路径字符串，该函数会自动递归到它的所有子目录中。第二个参数topdown默认为True表示首先返回目录下的文件，然后是目录。False表示首先返回目录列表，然后是文件列表。第三个参数onerror可以设为函数，以在遇到错误时发生什么。

二、使用os.walk()函数

下面是一个简单的使用os.walk()函数的示例，它打印出从一个指定的目录开始的所有子目录中的所有文件名和目录名。

import os

for root, dirs, files in os.walk("/"):
    for file in files:
        print(os.path.join(root, file))

    for dir in dirs:
        print(os.path.join(root, dir))

运行此代码，将从“/”目录开始遍历整个系统的文件。如果希望仅遍历特定的目录，只需将os.walk()中的第一个参数更改为所需的目录路径。

三、os.walk()函数的应用

1）在指定目录中查找特定扩展名的文件

以下示例演示如何在指定目录中递归查找具有指定扩展名的所有文件并将它们打印出来。

import os

def find_files(directory, extension):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                print(os.path.join(root, file))

# Example
find_files('.', '.jpg')

2）在指定目录中查找特定文件名的文件

以下示例演示如何在指定目录中递归查找特定文件名的文件并将它们打印出来。

import os

def find_files(directory, filename):
    for root, dirs, files in os.walk(directory):
        if filename in files:
            print(os.path.join(root, filename))

# Example
find_files('.', 'index.html')

3）在指定目录中删除指定扩展名的文件

以下示例演示如何递归查找指定目录中扩展名为.txt的所有文件并将其删除。

import os

def delete_files(directory, extension):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                os.remove(os.path.join(root, file))

# Example
delete_files('.', '.txt')

总结

os.walk()是Python有用的文件和目录操作函数之一。使用os.walk()函数可以迭代遍历指定目录下的所有文件夹及其子文件夹。本文提供了os.walk()函数的详细介绍和代码示例，可供读者参考和练习。当您需要遍历指定目录时，使用os.walk()函数可以方便地解决问题。