Python遍历文件夹所有指定扩展名文件
在日常工作中，我们经常需要对特定类型的文件进行操作，比如图片、文档、视频等。而这些文件往往保存在不同的文件夹中，如果要对这些文件进行批量处理，就需要编写程序来遍历指定文件夹中的所有特定扩展名的文件。本文将介绍如何使用Python编写代码来实现这一功能。
准备工作
在开始编写代码之前，我们需要安装Python的os模块和glob模块。os模块提供了访问操作系统功能的接口，而glob模块则用于查找文件路径名匹配指定模式的所有文件。
import os
import glob
遍历文件夹
首先，我们需要定义一个函数来遍历指定文件夹中的所有文件，并返回所有符合条件的文件路径列表。
def find_files(dir_path, ext):
files_list = []
for root, dirs, files in os.walk(dir_path):
for file in files:
if file.endswith(ext):
file_path = os.path.join(root, file)
files_list.append(file_path)
return files_list
在这个函数中，我们使用os.walk()方法来遍历指定文件夹中的所有文件，然后使用endswith()方法来筛选出指定扩展名的文件，并将符合条件的文件路径添加到列表中。
使用glob模块查找指定扩展名文件
另一种方法是使用glob模块的glob()方法来查找指定扩展名的文件。
def find_files_glob(dir_path, ext):
search_pattern = os.path.join(dir_path, f'*.{ext}')
files_list = glob.glob(search_pattern)
return files_list
这个函数中，我们使用os.path.join()方法构建了一个搜索模式，然后使用glob.glob()方法查找所有符合条件的文件。
示例
现在，我们来看一个示例，假设我们要查找一个文件夹中的所有txt文件。
dir_path = 'path/to/your/folder'
ext = 'txt'
files_list = find_files(dir_path, ext)
print(files_list)
饼状图
接下来，我们使用饼状图来展示不同类型文件在文件夹中的比例。
pie
"PNG" : 30
"JPG" : 20
"TXT" : 10
"DOCX" : 15
状态图
最后，我们使用状态图来展示文件遍历的流程。
stateDiagram
[*] --> Start
Start --> FindFiles
FindFiles --> End
End --> [*]
总结
通过本文的介绍，我们学习了如何使用Python来遍历文件夹中所有指定扩展名的文件。我们介绍了两种方法，一种是使用os模块的os.walk()方法，另一种是使用glob模块的glob()方法。我们还展示了如何使用饼状图和状态图来辅助我们理解代码的执行流程。希望本文能帮助你更好地理解和应用Python中的文件操作技巧。
