python怎么遍历一个文件夹里所有的图片,并且根据不同大小进行分类,放到不同文件夹里
时间: 2023-03-28 17:00:29 浏览: 71
您好,可以使用Python中的os和PIL库来实现遍历文件夹里所有的图片,并根据不同大小进行分类,放到不同文件夹里。具体实现方法可以参考以下代码:
```
import os
from PIL import Image
# 定义分类函数
def classify_image(file_path):
with Image.open(file_path) as img:
width, height = img.size
if width > 100 or height > 100:
return 'large'
elif width > 500 or height > 500:
return 'medium'
else:
return 'small'
# 遍历文件夹
for root, dirs, files in os.walk('your_folder_path'):
for file in files:
if file.endswith('.jpg') or file.endswith('.png'):
file_path = os.path.join(root, file)
category = classify_image(file_path)
# 根据分类结果创建对应的文件夹
if not os.path.exists(category):
os.makedirs(category)
# 将文件移动到对应的文件夹
os.rename(file_path, os.path.join(category, file))
```
以上代码中,`classify_image`函数用于根据图片大小分类,返回分类结果,`os.walk`函数用于遍历文件夹,`os.path.join`函数用于拼接路径,`os.makedirs`函数用于创建文件夹,`os.rename`函数用于移动文件。