在python中为图像自动创建文件夹

import os
import shutil

image_dir = "指定目录路径"
image_files = [f for f in os.listdir(image_dir) if f.endswith((".jpg", ".jpeg", ".png"))]

for image_file in image_files:
    file_ext = os.path.splitext(image_file)[1]
    folder_name = file_ext[1:].upper() + "_Images"  # 创建的文件夹名，如 JPG_Images
    folder_path = os.path.join(image_dir, folder_name)  # 文件夹的完整路径
    os.makedirs(folder_path, exist_ok=True)  # 创建文件夹，如果已存在则不会报错

    image_path = os.path.join(image_dir, image_file)  # 图像文件的完整路径
    shutil.move(image_path, folder_path)  # 移动图像文件到对应的文件夹中