按文件名对文件夹中的文件分类

我的文件夹中的部分文件，需要同一类的图片放在同一文件夹下（obj后面是类名）。

#用于获取文件夹中的文件对象、创建文件夹
import shutil
#移动文件到指定文件夹中
import os
 
path = './';
files_list = os.listdir(path)
for file in files_list:
	#分离文件名和扩展名
    filename, suffix = os.path.splitext(file)
    data = filename.split('_')[0];
    #选出作为文件夹名的部分
    res = data[3:];
    #新的文件夹的路径
    folder_name = path+res;
    if not os.path.exists(folder_name):
         os.makedirs(folder_name)
    #移动文件到指定文件夹
    shutil.move(file, folder_name)

分类后得到的文件夹结构：