### Python 根据文件扩展名筛选文件的方法 在 Python 中，可以通过多种方式根据文件扩展名筛选文件。以下是几种常见且有效的方法： #### 方法一：使用 `os` 模块 通过 `os.listdir()` 方法获取目录下的所有文件，并结合 `os.path.splitext()` 函数提取文件的扩展名进行筛选[^1]。 ```python import os folder_path = "/path/to/folder" # 替换为实际路径 extension = ".py" # 指定要筛选的文件扩展名 for filename in os.listdir(folder_path): if os.path.splitext(filename)[1] == extension: # 判断文件扩展名是否匹配 print(os.path.join(folder_path, filename)) ``` 上述代码中，`os.listdir(folder_path)` 返回指定目录下的所有文件和文件夹名称列表[^1]。`os.path.splitext(filename)[1]` 提取文件扩展名并进行比较，从而筛选出符合要求的文件。 #### 方法二：使用 `glob` 模块 `glob` 模块提供了一种简单的方式来查找符合特定模式的文件。例如，可以使用通配符 `*` 来匹配具有指定扩展名的所有文件[^4]。 ```python import glob import os folder_path = "/path/to/folder" # 替换为实际路径 extension = "*.py" # 指定要筛选的文件扩展名 file_list = glob.glob(os.path.join(folder_path, extension)) # 匹配指定扩展名的文件 for file in file_list: print(file) ``` `glob.glob()` 函数支持使用通配符来匹配文件名，因此可以直接指定扩展名作为匹配条件[^4]。这种方法比手动遍历文件列表更加简洁高效。 #### 方法三：使用 `pathlib` 模块 `pathlib` 模块提供了面向对象的方式处理文件系统路径。通过 `Path.rglob()` 或 `Path.glob()` 方法可以轻松筛选具有特定扩展名的文件[^3]。 ```python from pathlib import Path folder_path = "/path/to/folder" # 替换为实际路径 extension = "*.py" # 指定要筛选的文件扩展名 path = Path(folder_path) file_list = path.rglob(extension) # 递归查找指定扩展名的文件 for file in file_list: print(file) ``` `Path.rglob()` 方法递归地搜索子目录中的所有文件，而 `Path.glob()` 仅搜索当前目录下的文件[^3]。这种方法更现代化且易于理解。 #### 方法四：结合命令行参数实现灵活筛选 如果需要通过命令行传递文件扩展名作为参数，可以使用 `argparse` 模块实现动态筛选。 ```python import os import argparse parser = argparse.ArgumentParser(description="筛选指定扩展名的文件") parser.add_argument("folder", help="目标文件夹路径") parser.add_argument("-e", "--extension", required=True, help="文件扩展名") args = parser.parse_args() folder_path = args.folder extension = args.extension for filename in os.listdir(folder_path): if os.path.splitext(filename)[1] == extension: # 判断文件扩展名是否匹配 print(os.path.join(folder_path, filename)) ``` 运行脚本时，可以通过命令行传递目标文件夹路径和扩展名参数，例如： ```bash python script.py /path/to/folder -e .py ``` ### 总结 以上方法展示了如何在 Python 中根据文件扩展名筛选文件。`os` 模块适用于基本操作，`glob` 模块适合使用通配符匹配，`pathlib` 模块则提供了更现代化的接口，而结合 `argparse` 可以实现灵活的命令行交互。