如何在Python中列出子目录并管理文件结构

在Python中，列出子目录可以通过内置的`os`模块或`pathlib`模块来实现。这里介绍一些常用的方法，从简单到复杂排序推荐，帮助你轻松获取所需的目录列表。

1. 使用os模块列出子目录

使用`os.listdir()`函数是最简单的方法之一。这个函数将返回指定路径中所有文件和目录的列表。我们可以结合列表解析和`os.path.isdir()`来筛选出子目录。

import os

def list_subdirectories(path):
    return [d for d in os.listdir(path) if os.path.isdir(os.path.join(path, d))]

subdirs = list_subdirectories('/path/to/directory')  # 替换成你的路径
print(sorted(subdirs))

2. 使用os.walk()列出子目录

`os.walk()`提供了一种遍历目录树的方法，可以列出所有子目录及其子目录。这种方法适合需要获取深层次子目录结构的情况。

import os

def list_all_subdirectories(path):
    subdirs = []
    for root, dirs, files in os.walk(path):
        for dir in dirs:
            subdirs.append(os.path.join(root, dir))
    return subdirs

all_subdirs = list_all_subdirectories('/path/to/directory')
print(sorted(all_subdirs))

3. 使用pathlib模块列出子目录

在Python 3.4及以上版本，建议使用`pathlib`模块。这个模块提供了面向对象的路径操作方式，可以更直观地处理文件和目录。

from pathlib import Path

def list_subdirectories_pathlib(path):
    return [d for d in Path(path).iterdir() if d.is_dir()]

subdirs = list_subdirectories_pathlib('/path/to/directory')
print(sorted(subdirs))

4. 自定义过滤条件列出子目录

有时候你可能只对某些特定格式的子目录感兴趣，比如以特定字符开头的目录。这时可以通过自定义过滤函数来实现。

def list_filtered_subdirectories(path, filter_str):
    return [d for d in os.listdir(path) if os.path.isdir(os.path.join(path, d)) and d.startswith(filter_str)]

filtered_subdirs = list_filtered_subdirectories('/path/to/directory', 'prefix_')
print(sorted(filtered_subdirs))

问答环节

使用os模块列出子目录时，如何判断一个路径是目录?

使用`os.path.isdir()`函数，可以判断给定路径是否为一个目录。如果是目录，该函数返回True，否则返回False。这是确保我们只列出目录而非文件的关键。

os.walk()方法的输出结构是什么样的?

`os.walk()`返回一个生成器，每次循环生成三个值：当前路径、当前路径下的子目录列表和当前路径下的文件列表。通过这三个值结合使用，可以获取深层次的目录信息。

如何使用pathlib模块列出特定后缀名的子目录?

使用`Path.glob()`方法可以实现对特定后缀名的搜索。例如，使用`Path(path).glob(‘*.py’)`可以列出所有Python文件，但通常目录不带有后缀，所以针对目录可以使用`iterdir()`方法并手动判断或组合自己的条件。