如何根据Python中列表中的文件名将图像文件分类到两个文件夹中？

import os
import shutil

# 假设这是你的文件名列表
file_names = ['image1.jpg', 'image2.png', 'image3.jpg', 'image4.png']

# 定义两个目标文件夹
folder_A = 'folder_A'
folder_B = 'folder_B'

# 创建目标文件夹（如果不存在）
if not os.path.exists(folder_A):
    os.makedirs(folder_A)
if not os.path.exists(folder_B):
    os.makedirs(folder_B)

# 遍历文件名列表并进行分类
for file_name in file_names:
    if file_name.startswith('image1'):
        shutil.move(file_name, os.path.join(folder_A, file_name))
    elif file_name.startswith('image2'):
        shutil.move(file_name, os.path.join(folder_B, file_name))

print("文件分类完成！")