在 Python 中，可以通过 `os` 模块和 `pathlib` 模块实现对目录中文件的查找与筛选。以下是一些常见的方法，可以根据文件名、扩展名、文件内容等条件进行筛选。 ### 使用 `os.listdir()` 筛选文件 `os.listdir()` 函数可以列出指定目录下的所有文件和子目录名。通过遍历这些文件名，并结合条件判断，可以筛选出符合条件的文件。 例如，筛选文件名中包含特定字符串的文件： ```python import os directory = '/path/to/directory' # 筛选文件名中包含 'example' 的文件 for filename in os.listdir(directory): if 'example' in filename: print(os.path.join(directory, filename)) ``` ### 使用 `os.walk()` 遍历目录并筛选文件 如果需要递归地查找子目录中的文件，可以使用 `os.walk()`。该函数会遍历指定目录及其所有子目录。 ```python import os directory = '/path/to/directory' # 查找所有 .txt 文件 for root, dirs, files in os.walk(directory): for file in files: if file.endswith('.txt'): print(os.path.join(root, file)) ``` ### 使用 `pathlib` 模块筛选文件 Python 3.4 引入了 `pathlib` 模块，它提供了一种面向对象的方式来处理文件路径，使用起来更加直观。 ```python from pathlib import Path directory = Path('/path/to/directory') # 查找所有 .py 文件 for file in directory.glob('**/*.py'): print(file) ``` ### 根据文件名的特定格式筛选文件 有时文件名具有特定格式，例如以某个字符串开头或中间包含特定标识，可以通过字符串操作进行筛选。 ```python import os directory = '/path/to/directory' # 筛选文件名第四个字符后包含 'target' 的文件 for filename in os.listdir(directory): if filename.split('_')[0] == 'target': print(os.path.join(directory, filename)) ``` ### 结合 Pandas 进行更复杂的筛选 如果文件名需要根据某个数据表（如 Excel 文件）中的信息进行匹配，可以结合 `pandas` 进行处理。 ```python import os import pandas as pd import shutil file_path = r'/path/to/files' save_path = r'./test' df = pd.read_excel(r'./1.xlsx') for file in os.listdir(file_path): for index, row in df.iterrows(): if str(row['FID']) == file.split('_')[4]: shutil.copy(os.path.join(file_path, file), os.path.join(save_path, file)) ``` ### 根据文件内容筛选文件 如果需要根据文件内容筛选文件，可以逐行读取文件内容，并检查是否满足特定条件。 ```python import os directory = '/path/to/directory' # 查找包含特定字符串的文本文件 for filename in os.listdir(directory): if filename.endswith('.txt'): with open(os.path.join(directory, filename), 'r') as file: if 'target_string' in file.read(): print(os.path.join(directory, filename)) ``` ### 使用正则表达式进行复杂匹配 如果文件名的匹配规则较为复杂，可以使用 `re` 模块进行正则表达式匹配。 ```python import os import re directory = '/path/to/directory' # 匹配文件名中符合特定正则表达式的文件 pattern = re.compile(r'^example_\d+\.txt$') for filename in os.listdir(directory): if pattern.match(filename): print(os.path.join(directory, filename)) ```