python 当前目录下文件夹

原创

mob64ca12d70c79 2023-12-06 07:08:48 ©著作权

文章标签 当前目录 Python 流程图 文章分类 Python 后端开发

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

Python 当前目录下文件夹

在 Python 中，我们经常需要获取当前目录下的文件夹列表。这可以通过使用 os 模块和一些内置函数来实现。本文将为您详细介绍如何使用 Python 获取当前目录下的文件夹，并提供代码示例和流程图。

使用 os 模块

Python 中的 os 模块提供了许多与操作系统交互的函数。其中之一是 os.listdir() 函数，它返回指定目录下的所有文件和文件夹的列表。我们可以使用该函数获取当前目录下的文件夹列表。

以下是获取当前目录下文件夹的示例代码：

import os

def get_folders_in_current_directory():
    folders = []
    for item in os.listdir():
        if os.path.isdir(item):
            folders.append(item)
    return folders

folders = get_folders_in_current_directory()
print(folders)

在上面的代码中，我们定义了一个名为 get_folders_in_current_directory() 的函数，该函数使用 os.listdir() 循环遍历当前目录中的所有项目。如果项目是文件夹，则将其添加到 folders 列表中。最后，函数返回 folders 列表。

流程图

下面是获取当前目录下文件夹的流程图：

graph TB
    A[开始] --> B[获取当前目录下的所有项目]
    B --> C{项目是否为文件夹？}
    C -- 是 --> D[将文件夹添加到列表中]
    C -- 否 --> B
    D --> B
    B --> E[返回文件夹列表]
    E --> F[结束]

上面的流程图使用了 mermaid 语法的 flowchart TD。

示例代码

下面是一个完整的示例代码，演示了如何使用 Python 获取当前目录下的文件夹并打印出来：

import os

def get_folders_in_current_directory():
    folders = []
    for item in os.listdir():
        if os.path.isdir(item):
            folders.append(item)
    return folders

folders = get_folders_in_current_directory()
print(folders)

运行上述代码，将会打印出当前目录下的所有文件夹的列表。