使用 Python 遍历指定后缀的文件

作为一名刚入行的小白，学习如何使用 Python 遍历当前文件夹下指定后缀的文件是一个基本而重要的技能。本文将为你提供一个详细的流程，以便你能够轻松地实现这一功能。

流程概述

首先，我们来看看整个流程的步骤。以下是一个简洁的流程表，以帮助你理解每一步。

步骤 描述
1 导入必要的库
2 获取当前工作目录
3 遍历当前目录的文件
4 检查文件后缀
5 打印符合条件的文件

下面是使用 Mermaid 语法表示的流程图：

flowchart TD
    A[开始] --> B[导入必要的库]
    B --> C[获取当前工作目录]
    C --> D[遍历当前目录的文件]
    D --> E[检查文件后缀]
    E --> F[打印符合条件的文件]
    F --> G[结束]

每一步的代码实现

1. 导入必要的库

在 Python 中，我们通常需要用到 os 和 fnmatch 模块来处理文件和目录。

import os  # 用于与操作系统交互
import fnmatch  # 用于文件名模式匹配

2. 获取当前工作目录

我们使用 os.getcwd() 方法来获取当前工作目录，这将帮助我们遍历该目录内的所有文件。

current_directory = os.getcwd()  # 获取当前工作目录

3. 遍历当前目录的文件

使用 os.listdir() 来获取当前目录下所有文件和文件夹的列表，并进行遍历。

files = os.listdir(current_directory)  # 列出当前目录下所有文件和文件夹

4. 检查文件后缀

为了筛选出特定后缀的文件，比如 .txt，我们可以使用 fnmatch 来进行匹配。

target_extension = '*.txt'  # 设置目标后缀
for file in files:  # 遍历文件列表
    if fnmatch.fnmatch(file, target_extension):  # 判断文件是否符合后缀

5. 打印符合条件的文件

如果文件符合指定的后缀，我们就打印出符合条件的文件名。

        print(file)  # 打印出符合条件的文件名

完整代码示例

将上述代码整合在一起，形成一个完整的 Python 脚本如下：

import os  # 用于与操作系统交互
import fnmatch  # 用于文件名模式匹配

current_directory = os.getcwd()  # 获取当前工作目录
files = os.listdir(current_directory)  # 列出当前目录下所有文件和文件夹
target_extension = '*.txt'  # 设置目标后缀

for file in files:  # 遍历文件列表
    if fnmatch.fnmatch(file, target_extension):  # 判断文件是否符合后缀
        print(file)  # 打印出符合条件的文件名

甘特图表示任务时间安排

以下是任务的甘特图，展示我们完成这个项目各个任务的时间安排：

gantt
    title 遍历文件任务
    dateFormat  YYYY-MM-DD
    section 任务
    导入必要的库: 2023-10-01, 1d
    获取当前工作目录: 2023-10-02, 1d
    遍历文件: 2023-10-03, 1d
    检查后缀: 2023-10-04, 1d
    打印结果: 2023-10-05, 1d

结尾

通过本文的介绍，你已经了解了如何使用 Python 遍历当前文件夹下指定后缀的文件。掌握这一技能后，你可以根据不同的需求进行更复杂的文件操作。希望你能不断实践和探索，成为一名优秀的开发者！