在Python中处理文件时，查找具有特定扩展名的文件是一项常见任务。本文探讨了几种高效且通用的方法来实现此目标，以满足不同的需求和编码风格。我们将介绍各种Python模块和技术，使您可以为您的特定用例选择最佳方法。
目录
1. 使用glob模块
glob模块提供了一种简洁的方法来查找单个目录中与定义模式匹配的文件。其glob()函数返回满足模式的路径列表。
import glob
def find_files_with_extension_glob(directory, extension):
"""使用glob.glob()查找具有给定扩展名的文件。"""
return glob.glob(f"{directory}/*{extension}")
# 示例：
files = find_files_with_extension_glob("./my_directory", ".txt")
print(files)
2. 利用os模块
os模块提供了更细粒度的控制。os.listdir()列出目录中的所有条目，允许自定义过滤。
import os
def find_files_with_extension_os(directory, extension):
"""使用os.listdir()查找具有给定扩展名的文件。"""
files = []
for filename in os.listdir(directory):
if filename.endswith(extension):
files.append(os.path.join(directory, filename))
return files
# 示例：
files = find_files_with_extension_os("./my_directory", ".txt")
print(files)
3. 使用pathlib进行单目录搜索
pathlib模块提供了一种面向对象的方法。其glob()方法比glob.glob()提供了更简洁的语法。
from pathlib import Path
def find_files_with_extension_pathlib(directory, extension):
"""使用pathlib.glob()查找具有给定扩展名的文件。"""
return list(Path(directory).glob(f"*{extension}"))
# 示例：
files = find_files_with_extension_pathlib("./my_directory", ".txt")
print(files)
4. 使用os.walk进行递归搜索
对于搜索子目录，os.walk()遍历目录树，允许您检查每个文件是否具有所需的扩展名。
import os
def find_files_recursive_os(directory, extension):
"""使用os.walk()递归查找具有给定扩展名的文件。"""
files = []
for root, _, filenames in os.walk(directory):
for filename in filenames:
if filename.endswith(extension):
files.append(os.path.join(root, filename))
return files
# 示例：
files = find_files_recursive_os("./my_directory", ".txt")
print(files)
5. 使用pathlib.rglob进行递归搜索
pathlib的rglob()方法为递归搜索提供了最优雅的解决方案。
from pathlib import Path
def find_files_recursive_pathlib(directory, extension):
"""使用pathlib.rglob()递归查找具有给定扩展名的文件。"""
return list(Path(directory).rglob(f"*{extension}"))
# 示例：
files = find_files_recursive_pathlib("./my_directory", ".txt")
print(files)
请记住创建一个包含一些.txt文件的“my_directory”文件夹才能使示例正常工作。选择最符合您的编码风格和项目需求的方法。对于简单的单目录搜索，glob或pathlib.glob()是不错的选择。对于递归搜索，pathlib.rglob()提供了最简洁易读的解决方案。