Python 中 os 模块仅读遍历文件夹
在日常的编程工作中，我们经常需要对文件系统进行操作，特别是遍历文件夹以获取文件列表。Python 的 os 模块提供了一些功能来帮助我们实现这一点。本文将重点介绍如何使用 Python 的 os 模块仅读遍历文件夹，并通过代码示例阐明其具体操作。同时，我们将通过流程图和关系图来理清遍历文件夹的过程和相关概念。
什么是 os 模块？
os 模块是 Python 标准库的一部分，它提供了一系列与操作系统交互的功能。通过 os 模块，我们可以访问文件系统中的文件和目录，执行文件操作，以及获取系统信息等。
遍历文件夹
遍历文件夹的常用方法有两种：使用 os.listdir() 和 os.walk()。其中，os.listdir() 方法适合用来获取指定目录下的所有文件和子目录，而 os.walk() 方法则可以递归遍历目录及其所有子目录。
示例代码
下面的代码示例展示了如何使用 os.listdir() 和 os.walk() 来遍历文件夹：
import os
# 使用 os.listdir() 遍历文件夹
def list_directory(path):
try:
print(f"Listing files in '{path}':")
for item in os.listdir(path):
print(item)
except FileNotFoundError:
print("Directory not found.")
# 使用 os.walk() 遍历文件夹，包括子目录
def walk_directory(path):
try:
print(f"Walking through the directory '{path}':")
for root, dirs, files in os.walk(path):
print(f"Current directory: {root}")
for file in files:
print(f" File: {file}")
except FileNotFoundError:
print("Directory not found.")
# 示例调用
list_directory('/path/to/your/directory')
walk_directory('/path/to/your/directory')
在这个示例中，list_directory 函数将列出指定路径中的文件，而 walk_directory 函数将递归遍历该路径及其所有子目录，输出每个文件的名称和所在目录。
程序的工作流程
为了更好地理解上述代码的工作流程，我们将其流程化并绘制成一个流程图：
flowchart TD
A[开始] --> B{是否存在该目录？}
B -- 是 --> C[列出目录下文件]
B -- 否 --> D[输出“目录未找到”]
C --> E[结束]
D --> E
代码解析
- 导入模块：首先，我们导入
os模块，这是我们进行文件遍历的依赖库。 - 定义函数：我们分别定义了
list_directory和walk_directory函数，它们接收一个文件夹路径作为参数。 - 异常处理：在这两个函数中，我们使用
try...except语句捕获可能出现的文件未找到错误。 - 遍历文件：
- 对于
os.listdir()，我们简单地列出文件夹中的所有文件和子目录。 - 对于
os.walk()，我们使用一个多重循环来遍历每一个子目录，获取当前目录、所有子目录和文件。
- 对于
关系图
为了进一步理解 os 模块与文件系统的关系，我们可以用 ER 图展示它们之间的关系：
erDiagram
DIR {
string name
string path
}
FILE {
string name
string extension
}
DIR ||--o{ FILE : contains
DIR ||--o{ DIR : contains
在该关系图中，每个目录（DIR）可以包含多个文件（FILE）和子目录（DIR），这形象地表示了目录结构的层级关系。
结论
通过浏览上述代码和流程图，我们可以看到如何利用 Python 的 os 模块方便地遍历文件夹。无论是简单的文件列出，还是复杂的递归遍历，在 Python 中都可以用简洁的代码轻松实现。掌握这些基本的文件操作将为我们后续的编程打下坚实的基础。在实际应用中，开发者可以灵活地根据需求选择合适的方法，从而高效地管理和操作文件系统。希望这篇文章能够帮助你更好地理解和运用 Python 的文件管理功能。
