python脚本所在目录

原创

mob64ca12e08acf 2023-10-02 04:22:34 ©著作权

文章标签 文件列表 python 绝对路径 文章分类 Python 后端开发

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

从python脚本所在目录获取文件列表的方法

在Python中，我们经常需要获取当前脚本所在目录的文件列表。这个需求可能是为了读取特定类型的文件，或者为了对目录中的文件进行某些操作。本文将介绍如何通过python脚本所在目录获取文件列表的方法，并提供相应的代码示例。

首先，我们需要导入os模块，它提供了与操作系统相关的功能，包括文件和目录操作。

import os

然后，我们可以使用os.path模块中的abspath()函数获取当前脚本的绝对路径。绝对路径是从根目录开始的完整路径，可以唯一地标识一个文件或目录。

script_dir = os.path.dirname(os.path.abspath(__file__))

接下来，我们可以使用os.listdir()函数获取脚本所在目录的文件列表。这个函数返回一个包含目录中所有文件和子目录的列表。

file_list = os.listdir(script_dir)

为了筛选出特定类型的文件，可以使用字符串的endswith()方法。这个方法返回一个布尔值，指示字符串是否以指定的后缀结尾。我们可以将这个方法应用于文件列表中的每个文件名。

下面是一个示例代码，它获取当前脚本所在目录的所有.txt文件。

txt_files = [f for f in file_list if f.endswith('.txt')]

现在，我们已经得到了当前脚本所在目录中所有.txt文件的列表，可以对它们进行进一步的操作，比如读取文件内容或对文件进行处理。

下面是一个完整的示例代码，它获取当前脚本所在目录的.txt文件，并打印出每个文件的名称和大小。

import os

script_dir = os.path.dirname(os.path.abspath(__file__))
file_list = os.listdir(script_dir)
txt_files = [f for f in file_list if f.endswith('.txt')]

for file_name in txt_files:
    file_path = os.path.join(script_dir, file_name)
    file_size = os.path.getsize(file_path)
    print(f"File: {file_name}, Size: {file_size} bytes")

通过以上代码，我们可以轻松地获取当前脚本所在目录的文件列表，并对特定类型的文件进行操作。

序列图

下面是一个使用mermaid语法中的sequenceDiagram标识的序列图。它展示了从获取当前脚本所在目录到获取文件列表的过程。

sequenceDiagram
    participant PythonScript as Python脚本
    participant OSModule as os模块
    participant FilePath as 文件路径
    participant FileList as 文件列表

    PythonScript ->> OSModule: 导入os模块
    PythonScript ->> OSModule: 获取脚本所在目录
    OSModule ->> FilePath: 获取脚本的绝对路径
    PythonScript ->> OSModule: 获取文件列表
    OSModule ->> FileList: 返回文件列表
    PythonScript ->> FilePath: 筛选文件类型
    FilePath ->> FileList: 返回符合条件的文件列表
    PythonScript -->> FileList: 对文件列表进行操作

通过这个序列图，我们可以更加直观地理解从获取当前脚本所在目录到获取文件列表的过程。