使用 Python 读取文件名并存入数组
在很多数据处理和应用程序开发的场景下，我们常常需要读取一个目录中的所有文件。这可以用于数据采集、文件处理、数据分析等多种任务。在 Python 中，我们可以很方便地使用标准库中的模块来完成这个任务。本文将详细介绍如何借助 Python 读取一个指定目录下的全部文件名，并将这些文件名存入数组中。
一、Python 的基础概述
Python 是一种广泛使用的高级编程语言，它以简洁、易读的语法而闻名。Python 提供了大量的标准库，使得开发者能很方便地进行文件操作、数据分析等任务。在本篇文章中，我们将主要聚焦于文件系统的操作。
二、读取目录中的文件名
为了读取指定目录中的所有文件名，我们可以使用 os 模块中的 listdir 方法或 os.walk 方法。下面是一个简单的示例，展示如何使用这两种方法。
1. 使用 os.listdir
os.listdir 方法可以列出指定目录下的所有文件和子目录，但不包含子目录中的文件。下面是一个示例代码：
import os
# 指定要读取的目录
directory = 'your_directory_path_here'
# 使用 os.listdir 获取目录下的所有文件和子目录
files_and_dirs = os.listdir(directory)
# 过滤出文件名
files = [f for f in files_and_dirs if os.path.isfile(os.path.join(directory, f))]
print(files)
2. 使用 os.walk
使用 os.walk 方法则可以遍历指定目录下的所有文件，包括子目录中的文件。下面是使用 os.walk 的示例代码：
import os
# 指定要读取的目录
directory = 'your_directory_path_here'
# 初始化一个空数组用于存储文件名
file_names = []
# 使用 os.walk 遍历目录
for root, dirs, files in os.walk(directory):
for file in files:
file_names.append(file)
print(file_names)
三、代码解析
在上述代码中，我们首先导入了 os 模块。然后通过 os.listdir 方法获取指定目录下的所有文件和子目录。在第二个示例中，我们通过 os.walk 方法遍历目录，它返回三个值：当前路径（root）、当前路径下的目录名（dirs）、当前路径下的文件名（files）。
3. 处理文件名
您可能会想进一步处理这些文件名，例如读取文件内容，分析文件类型或重命名文件等。这时候，可以很方便地对 files 或 file_names 这两个数组进行处理。例如，我们可以打印出文件的完整路径：
for file in file_names:
full_path = os.path.join(directory, file)
print(full_path)
四、使用示例
假设我们有一个目录结构如下：
your_directory_path_here/
│
├── file1.txt
├── file2.txt
└── subdirectory/
├── file3.txt
└── file4.txt
通过上述代码，我们可以分别获取 your_directory_path_here 目录下的文件名和所有子目录中的文件名。
五、总结与应用
在本文中，我们探讨了如何使用 Python 中的 os 模块读取目录中的文件名，并将这些文件名存入数组。无论是数据分析、文件管理还是自动化脚本，这一技术都非常有用。
当然，除了读取文件名之外，Python 提供了很多其他功能来处理文件，如读取文件内容、写入文件、新建目录、删除文件等。了解这些功能将极大地提升您的生产力和工作效率。
在结束之前，让我们回顾一下上述代码与数据的关系，利用 ER 图来表现这些关系：
erDiagram
Directory {
string path
}
File {
string name
string extension
string content
}
Directory ||..|| File : contains
在这个关系图中，我们可以看到一个目录（Directory）中可以包含多个文件（File），文件具有名称、扩展名和内容等属性。
希望本文的介绍能帮助你更好地理解如何利用 Python 进行文件操作，如果你在实践中遇到任何问题，欢迎随时向我提问！
