以下是一个用Python实现遍历目录并根据文件扩展名进行结果过滤的示例代码:
import os
def filter_files(directory, extension):
filtered_files = []
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith(extension):
filtered_files.append(os.path.join(root, file))
return filtered_files
directory = "/path/to/directory" # 替换为要遍历的目录路径
extension = ".txt" # 替换为要过滤的文件扩展名
filtered_files = filter_files(directory, extension)
for file in filtered_files:
print(file)
上述示例代码中,filter_files
函数接收一个目录路径和文件扩展名作为参数。它使用os模块中的walk
函数遍历目录,获取目录下的所有文件,并通过文件扩展名进行过滤。如果文件的扩展名与给定的扩展名相同,就将文件的完整路径添加到filtered_files
列表中。最后,通过遍历filtered_files
列表打印出所有符合条件的文件路径。
注意替换示例代码中的directory
和extension
变量为你自己的目录和扩展名。
下一篇:遍历一个目录路径的列表