如何使用Python获取目录所有文件夹名

总体流程

flowchart TD
    A[开始] --> B[导入os模块]
    B --> C[定义函数获取文件夹名]
    C --> D[遍历目录]
    D --> E[判断是否为文件夹]
    E --> F[获取文件夹名]
    F --> G[保存文件夹名列表]
    G --> H[返回文件夹名列表]
    H --> I[结束]

步骤及代码示例

  1. 导入os模块
import os
  1. 定义函数获取文件夹名
def get_folder_names(directory):
    folder_names = []  # 用于存放文件夹名的列表
    
    # 遍历目录
    for root, dirs, files in os.walk(directory):
        for d in dirs:
            folder_names.append(d)  # 将文件夹名添加到列表中
    
    return folder_names
  1. 调用函数并打印结果
directory = "/path/to/directory"  # 替换为你要获取文件夹名的目录路径
folders = get_folder_names(directory)
print(folders)

代码解释

  • os.walk(directory)：遍历指定目录下的所有文件和子目录。
  • for root, dirs, files in os.walk(directory):：遍历目录时，root表示当前目录路径，dirs表示当前目录下的子目录列表，files表示当前目录下的文件列表。
  • for d in dirs:：遍历子目录列表，即遍历当前目录下的所有文件夹。
  • folder_names.append(d)：将文件夹名添加到folder_names列表中。
  • return folder_names：返回包含所有文件夹名的列表。

饼状图

pie
    title 文件夹占比
    "文件夹1" : 30
    "文件夹2" : 20
    "文件夹3" : 50

通过上述步骤和代码示例，你可以轻松地使用Python获取指定目录下所有文件夹的名字。希望这篇文章对你有所帮助，祝你编程学习顺利！