python,写一个统计图片个数的函数,遍历文件夹中所有文件,统计图片数量,输出每个文件夹中有多少图片,图片的格式是什么
时间: 2025-06-29 09:12:28 浏览: 13
### 编写Python函数遍历文件夹并统计图片
为了满足需求,下面展示了一个Python函数`count_images_in_directory`,此函数能够递归地遍历给定路径下的所有子目录,并专门针对常见的图像文件格式(如`.jpg`, `.png`, 和`.gif`)进行计数。通过字典结构保存不同类型的图片数目以便于后续处理或显示。
```python
import os
def count_images_in_directory(directory_path):
image_extensions = ['.jpg', '.jpeg', '.png', '.gif'] # 定义要查找的图片扩展名列表
counts = {} # 创建一个空字典用于记录各类型图片的数量
for root, _, filenames in os.walk(directory_path): # 遍历目录树
for filename in filenames:
ext = os.path.splitext(filename)[1].lower() # 获取文件扩展名并转换成小写字母
if ext in image_extensions: # 判断是否为所需图片格式之一
relative_folder = os.path.relpath(root, directory_path) # 记录相对路径
if relative_folder not in counts:
counts[relative_folder] = {ext: 0 for ext in image_extensions} # 初始化当前文件夹内的计数字典
counts[relative_folder][ext] += 1 # 对应格式计数加一
return counts # 返回最终的结果字典
```
上述代码实现了对特定目录及其子目录内图片文件(.jpg,.jpeg,.png,.gif)数量的统计工作[^4]。每当遇到符合条件的文件时,都会更新对应文件夹下相应格式的计数值;最后返回的是一个嵌套字典形式的数据集,外层键表示各个不同的文件夹位置,而内部则包含了该文件夹中各种支持的图片格式的具体数目。
阅读全文
相关推荐
















