python遍历文件夹中子文件夹的数量,并判断各个子文件夹中的图片数量
时间: 2023-08-14 18:07:25 浏览: 117
可以使用Python中的os和os.path模块来实现遍历文件夹中子文件夹的数量,并使用Pillow库来判断各个子文件夹中的图片数量。以下是一个示例代码:
```python
import os
from PIL import Image
# 遍历文件夹
def traverse_folder(folder_path):
subfolders = 0
image_counts = []
for root, dirs, files in os.walk(folder_path):
subfolders += len(dirs)
count = 0
for file in files:
# 判断文件是否为图片
if file.endswith('.jpg') or file.endswith('.jpeg') or file.endswith('.png'):
count += 1
image_counts.append(count)
return subfolders, image_counts
# 测试代码
if __name__ == '__main__':
folder_path = 'your_folder_path'
subfolders, image_counts = traverse_folder(folder_path)
print('子文件夹数量:', subfolders)
print('各子文件夹中的图片数量:', image_counts)
```
在上述示例代码中,我们使用了os.walk()函数来遍历文件夹中的子文件夹,并使用len()函数计算子文件夹的数量。在遍历文件夹中的每个文件时,我们使用了Pillow库中的Image.open()函数来判断文件是否为图片,然后计算每个子文件夹中图片的数量,并将其存储在一个列表中返回。最后,我们在主函数中调用traverse_folder()函数,并打印子文件夹数量和各个子文件夹中的图片数量。
阅读全文