Python os.walk：快速扫描文件夹并进行文件处理

小蓝 • 2024-12-04 19:15 • 编程

os.walk()是Python标准库中的一个函数，可以快速扫描文件夹及其子文件夹中的所有文件，并对这些文件进行处理。在本文中，我们将深入介绍如何使用os.walk()函数。

一、什么是os.walk()函数？

os.walk()函数返回一个三元组，包含一个正在处理的文件夹的父级目录、当前文件夹的名称以及当前文件夹中的文件和文件夹。通过递归处理，该函数可以得到整个文件夹的目录结构。

下面是一个简单的使用os.walk()函数的示例：

import os

# 遍历文件夹中的所有文件
for root, dirs, files in os.walk("文件夹路径"):
    for name in files:
        print(os.path.join(root, name))
    for name in dirs:
        print(os.path.join(root, name))

上述Python代码使用os.walk()函数遍历了”文件夹路径”中的所有文件和文件夹，并输出了它们的相对或绝对路径。

二、如何使用os.walk()函数进行文件处理？

1. 根据文件名进行筛选

通过os.walk()函数，可以快速找到指定文件夹及其子文件夹中的所有文件，然后进行过滤，只处理需要的文件。例如，如果要处理所有以.py为扩展名的文件，可以使用以下代码：

import os

for root, dirs, files in os.walk("文件夹路径"):
    for file in files:
        if file.endswith(".py"):
            filepath = os.path.join(root, file)
            # 处理文件内容
            with open(filepath) as f:
                # 打印文件名及文件内容
                print(f"处理文件 {filepath}")
                print(f.read())

以上代码实现了对指定文件夹及其子文件夹中的所有.py文件进行处理。

2. 对文件进行重命名

在使用os.walk()函数遍历文件夹时，我们可以通过os.rename()函数对文件进行重命名。下面是一个将所有.jpg图片文件重命名为.png格式的示例：

import os

for root, dirs, files in os.walk("文件夹路径"):
    for file in files:
        if file.endswith(".jpg"):
            origin_filepath = os.path.join(root, file)
            new_filepath = os.path.join(root, file[:-4] + ".png")
            os.rename(origin_filepath, new_filepath)

以上代码实现了将指定文件夹及其子文件夹中所有的.jpg图片文件重命名为.png格式。

3. 对文件内容进行修改

在使用os.walk()函数遍历文件夹时，我们可以通过open()函数读取文件内容并进行修改，然后通过open()函数重新写入文件。

下面是一个将指定文件夹及其子文件夹中所有.txt文件中的字符”a”替换为”b”的示例：

import os

for root, dirs, files in os.walk("文件夹路径"):
    for file in files:
        if file.endswith(".txt"):
            filepath = os.path.join(root, file)
            # 修改文件内容
            with open(filepath, "r") as f:
                content = f.read()
            content = content.replace("a", "b")
            with open(filepath, "w") as f:
                f.write(content)

以上代码实现了将指定文件夹及其子文件夹中所有的.txt文件中的字符”a”修改为”b”。

三、如何遍历文件夹下的所有子文件夹？

在使用os.walk()函数遍历文件夹时，如果需要遍历所有子文件夹，则可以设置topdown=False。使用这个参数可以先处理子文件夹，然后再处理父文件夹。

下面是一个遍历文件夹及其所有子文件夹的示例：

import os

for root, dirs, files in os.walk("文件夹路径", topdown=False):
    for name in files:
        print(os.path.join(root, name))
    for name in dirs:
        print(os.path.join(root, name))

以上代码实现了遍历指定文件夹及其所有的子文件夹，并输出文件夹和文件的路径。

四、总结

os.walk()函数是Python标准库中一个非常有用的函数，它可以快速扫描指定文件夹及其子文件夹中的所有文件，并对这些文件进行处理。通过本文的介绍，你可以轻松地掌握os.walk()函数的使用方法，实现文件夹操作。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/199223.html