如何根据数据框中的文件名拆分图像文件夹？

import os
import pandas as pd

# 假设数据框df已经存在，且包含一列名为'filename'的文件名数据
df = pd.DataFrame({'filename': ['image_001.jpg', 'image_002.png', 'image_003.jpg']})

# 定义拆分函数
def split_and_create_folders(df, column_name, folder_prefix):
    for index, row in df.iterrows():
        filename = row[column_name]
        # 根据文件名前缀拆分
        folder_name = filename.split('_')[0]  # 假设前缀是下划线前的部分
        folder_path = os.path.join(folder_prefix, folder_name)
        
        # 创建文件夹（如果不存在）
        if not os.path.exists(folder_path):
            os.makedirs(folder_path)
        
        # 移动文件到对应文件夹（假设文件在当前目录下）
        src_path = os.path.join(os.getcwd(), filename)
        dst_path = os.path.join(folder_path, filename)
        os.rename(src_path, dst_path)

# 调用拆分函数
split_and_create_folders(df, 'filename', 'image_folders')