python批量提取文件名

原创

mob64ca12f831ae 2023-10-22 13:51:51 ©著作权

文章标签 文件名文件列表 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f831ae的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python批量提取文件名

本文将介绍如何使用Python批量提取文件名。在日常工作中，我们经常需要对多个文件进行操作，例如批量处理文件、重命名文件等。而文件名是我们对文件进行操作的关键信息，因此提取文件名是一个很常见的需求。

在Python中，我们可以使用os模块提供的函数来实现文件名的提取。os模块是Python标准库中的一个模块，提供了访问操作系统功能的方法。下面我们将使用os模块的函数来演示如何批量提取文件名。

1. 获取文件列表

首先，我们需要获取文件列表。可以通过os.listdir()函数来获取指定目录下的所有文件和文件夹的名称。以下是一个示例代码：

import os

# 指定目录路径
dir_path = 'path/to/directory'

# 获取目录下的所有文件和文件夹的名称
file_list = os.listdir(dir_path)

# 打印文件列表
print(file_list)

上述代码中，我们首先指定了一个目录路径dir_path，然后使用os.listdir()函数获取该目录下的所有文件和文件夹的名称，将结果保存在file_list变量中。最后，我们打印了file_list变量的值，即文件列表。

2. 提取文件名

接下来，我们需要从文件列表中提取文件名。文件名是文件路径的最后一部分，可以使用os.path.basename()函数来提取。以下是一个示例代码：

import os

# 指定目录路径
dir_path = 'path/to/directory'

# 获取目录下的所有文件和文件夹的名称
file_list = os.listdir(dir_path)

# 提取文件名
file_names = [os.path.basename(file) for file in file_list]

# 打印文件名列表
print(file_names)

在上述代码中，我们使用列表推导式将文件列表中的每个文件路径都通过os.path.basename()函数提取出文件名，并保存在file_names列表中。最后，我们打印了file_names列表的值，即文件名列表。

3. 序列图

下面是一个使用Python批量提取文件名的示例序列图：

sequenceDiagram
    participant User
    participant PythonScript
    participant OSModule

    User->>PythonScript: 指定目录路径
    PythonScript->>OSModule: 调用os.listdir()函数
    OSModule->>PythonScript: 返回文件列表
    PythonScript->>OSModule: 调用os.path.basename()函数
    OSModule->>PythonScript: 返回文件名列表
    PythonScript->>User: 打印文件名列表

上述序列图展示了用户指定目录路径后，Python脚本通过调用os模块的函数获取文件列表，并通过os.path.basename()函数提取文件名，最后将文件名列表打印给用户。