python如何读取指定路径的文件夹名

Python读取指定路径的文件夹名的步骤非常简单和直接：使用os模块获取文件夹名、使用os.walk遍历目录树、结合pathlib模块进行路径操作。其中，os模块和pathlib模块是Python标准库的一部分，它们提供了丰富的文件和目录操作功能。接下来，我们将详细展开如何在Python中读取指定路径的文件夹名。

一、使用os模块获取文件夹名

Python的os模块提供了一系列用于操作文件和目录的功能。要读取指定路径的文件夹名，可以使用os.listdir()函数和os.path模块的方法。

1. 基本用法

os.listdir()函数返回指定目录中的所有条目，包括文件和文件夹。我们可以通过os.path.isdir()函数来判断一个条目是否是文件夹。

import os
def get_folder_names(path):
    try:
        folder_names = [name for name in os.listdir(path) if os.path.isdir(os.path.join(path, name))]
        return folder_names
    except FileNotFoundError:
        return "The specified path does not exist."
    except PermissionError:
        return "Permission denied to access the specified path."
示例用法
path = '/your/specified/path'
print(get_folder_names(path))

2. 处理错误和异常

在实际应用中，路径可能不存在或没有权限访问。我们可以使用try–except块捕获这些异常。

import os
def get_folder_names(path):
    try:
        folder_names = [name for name in os.listdir(path) if os.path.isdir(os.path.join(path, name))]
        return folder_names
    except FileNotFoundError:
        return "The specified path does not exist."
    except PermissionError:
        return "Permission denied to access the specified path."
示例用法
path = '/your/specified/path'
print(get_folder_names(path))

二、使用os.walk遍历目录树

os.walk()生成目录树下的所有文件名，它是一个生成器，返回一个三元组(dirpath, dirnames, filenames)。我们可以使用它来遍历指定路径下的所有目录。

1. 基本用法

import os
def get_all_folder_names(root_path):
    folder_names = []
    for dirpath, dirnames, filenames in os.walk(root_path):
        for dirname in dirnames:
            folder_names.append(os.path.join(dirpath, dirname))
    return folder_names
示例用法
root_path = '/your/specified/path'
print(get_all_folder_names(root_path))

2. 控制遍历深度

os.walk()默认会递归遍历所有子目录。我们可以通过控制递归深度来限制其遍历范围。

import os
def get_folder_names_limited_depth(root_path, max_depth):
    folder_names = []
    for dirpath, dirnames, filenames in os.walk(root_path):
        depth = dirpath[len(root_path):].count(os.sep)
        if depth < max_depth:
            for dirname in dirnames:
                folder_names.append(os.path.join(dirpath, dirname))
        else:
            del dirnames[:]  # 防止进一步递归
    return folder_names
示例用法
root_path = '/your/specified/path'
max_depth = 2
print(get_folder_names_limited_depth(root_path, max_depth))

三、结合pathlib模块进行路径操作

pathlib模块提供了面向对象的路径操作方法，它使得路径操作更加直观和简洁。

1. 基本用法

from pathlib import Path
def get_folder_names_pathlib(path):
    try:
        p = Path(path)
        folder_names = [x.name for x in p.iterdir() if x.is_dir()]
        return folder_names
    except FileNotFoundError:
        return "The specified path does not exist."
    except PermissionError:
        return "Permission denied to access the specified path."
示例用法
path = '/your/specified/path'
print(get_folder_names_pathlib(path))

2. 递归遍历目录

pathlib的rglob()方法可以递归遍历目录树。

from pathlib import Path
def get_all_folder_names_pathlib(root_path):
    p = Path(root_path)
    folder_names = [x for x in p.rglob('*') if x.is_dir()]
    return folder_names
示例用法
root_path = '/your/specified/path'
print(get_all_folder_names_pathlib(root_path))

3. 控制遍历深度

通过生成器表达式和条件判断，我们可以控制递归的深度。

from pathlib import Path
def get_folder_names_limited_depth_pathlib(root_path, max_depth):
    p = Path(root_path)
    folder_names = [x for x in p.rglob('*') if x.is_dir() and len(x.relative_to(root_path).parts) <= max_depth]
    return folder_names
示例用法
root_path = '/your/specified/path'
max_depth = 2
print(get_folder_names_limited_depth_pathlib(root_path, max_depth))