[教程]掌握Python高效读取指定文件夹下所有子文件夹的技巧

在Python中，读取指定文件夹下的所有子文件夹是一个常见的任务，尤其是在处理文件系统结构时。以下是一些高效读取指定文件夹下所有子文件夹的技巧。使用os模块Python的os模块提供了访问操作系统功能...

在Python中，读取指定文件夹下的所有子文件夹是一个常见的任务，尤其是在处理文件系统结构时。以下是一些高效读取指定文件夹下所有子文件夹的技巧。

使用os模块

Python的os模块提供了访问操作系统功能的接口，其中包括遍历文件夹的功能。

1. 使用os.listdir()

os.listdir()函数可以列出指定路径下的所有文件和文件夹。结合循环和条件判断，可以筛选出子文件夹。

import os
def list_subfolders(path): subfolders = [] for entry in os.listdir(path): if os.path.isdir(os.path.join(path, entry)): subfolders.append(entry) return subfolders
# 使用示例
path_to_directory = '/path/to/your/directory'
subfolders = list_subfolders(path_to_directory)
print(subfolders)

2. 使用os.walk()

os.walk()是一个更高级的函数，它可以遍历指定目录及其所有子目录，并返回一个三元组(root, dirs, files)。

import os
def list_subfolders_with_os_walk(path): subfolders = [] for root, dirs, files in os.walk(path): for dir in dirs: subfolders.append(os.path.join(root, dir)) return subfolders
# 使用示例
path_to_directory = '/path/to/your/directory'
subfolders = list_subfolders_with_os_walk(path_to_directory)
print(subfolders)

使用pathlib模块

Python 3.4及以上版本引入了pathlib模块，它提供了一种面向对象的方式来处理文件系统路径。

1. 使用Path.iterdir()

Path.iterdir()方法可以遍历指定路径下的所有文件和文件夹。

from pathlib import Path
def list_subfolders_with_pathlib(path): subfolders = [d for d in path.iterdir() if d.is_dir()] return [str(d) for d in subfolders]
# 使用示例
path_to_directory = Path('/path/to/your/directory')
subfolders = list_subfolders_with_pathlib(path_to_directory)
print(subfolders)

性能考虑

• 当处理大量文件或大型文件夹结构时，os.walk()通常比os.listdir()更高效，因为它在遍历过程中减少了不必要的文件系统调用。
• 使用pathlib模块通常比使用os模块更简洁，但性能可能略低。

总结

通过使用os模块的os.listdir()和os.walk()，或者pathlib模块的Path.iterdir()，你可以高效地读取指定文件夹下的所有子文件夹。选择哪种方法取决于你的具体需求和偏好。