python 获取当前路径的文件夹列表

Python 获取当前路径的文件夹列表

在Python中，要获取当前路径的文件夹列表，我们可以使用os模块提供的方法。os模块是Python标准库中的一个模块，它提供了与操作系统相关的功能和接口，包括文件和目录操作、进程管理、环境变量等。

获取当前路径

首先，我们需要获取当前正在运行的脚本所在的路径，也就是当前工作目录。可以使用os模块中的getcwd()方法来获取当前工作目录：

import os

current_directory = os.getcwd()
print("当前工作目录：", current_directory)

上述代码中，os.getcwd()返回一个字符串，表示当前工作目录的绝对路径。我们可以将其保存在一个变量中，以便后续使用。

列出文件夹

有了当前工作目录，我们可以使用os.listdir()方法来列出其中的文件和文件夹。该方法接受一个路径作为参数，并返回该路径下的所有文件和文件夹的名称列表：

import os

current_directory = os.getcwd()
folders = [f for f in os.listdir(current_directory) if os.path.isdir(os.path.join(current_directory, f))]

print("当前路径的文件夹列表：")
for folder in folders:
    print(folder)

上述代码中，我们使用了列表推导式来过滤出当前路径下的文件夹。os.path.isdir()方法用于判断给定路径是否是一个文件夹。我们使用os.path.join()方法将当前路径与文件夹名称拼接起来，再调用os.path.isdir()进行判断。

完整代码

下面是获取当前路径的文件夹列表的完整示例代码：

import os

def get_folder_list():
    current_directory = os.getcwd()
    folders = [f for f in os.listdir(current_directory) if os.path.isdir(os.path.join(current_directory, f))]
    return folders

if __name__ == "__main__":
    folder_list = get_folder_list()
    if len(folder_list) > 0:
        print("当前路径的文件夹列表：")
        for folder in folder_list:
            print(folder)
    else:
        print("当前路径下没有文件夹")

在上述代码中，我们将获取文件夹列表的逻辑封装在一个函数get_folder_list()中，并通过if __name__ == "__main__":判断是否直接运行该脚本。

注意事项

需要注意的是，os.listdir()方法返回的列表包含了当前路径下的所有文件和文件夹的名称，包括隐藏文件和系统文件。如果只想获取文件夹列表，可以使用上述代码中的方法进行筛选，或者使用os.scandir()方法，该方法返回一个迭代器，只包含目录项。

状态图

下面是一个使用mermaid语法绘制的状态图，描述了获取当前路径的文件夹列表的过程：

stateDiagram
    [*] --> 获取当前路径
    获取当前路径 --> 列出文件夹
    列出文件夹 --> [*]

在状态图中，[*]表示初始状态和结束状态，-->表示状态之间的跳转。

总结

本文介绍了如何使用Python获取当前路径的文件夹列表。通过使用os模块提供的方法，我们可以轻松地获取当前工作目录，并列出其中的文件夹。希望本文对你有所帮助！