python将包含JPG和XML混合文件的文件夹分类子文件夹

import os
import shutil
allfile=os.listdir('/home/classfication/new')
for i in allfile:
    # 遍历每个父文件夹
    dir='home/classification'+i
    JPG=dir+'/jpg'
    # 创建jpg类的子文件夹
    JPG=dir+'/jpg'
    isExists=os.path.exists(JPG)
    if not isExists:
        os.makedirs(JPG)
    # 创建xml类的子文件夹
    XML=dir+'/xml'
    isExists=os.path.exists(XML)
    if not isExists:
        os.makedirs(XML)
    thing=os.listdir(dir+'/')
    for j in thing:
        if j[-4:]=='.jpg':
            shutil.move(dir+'/'+j,JPG+'/'+j)
        elif j[-4:]=='.xml':
            shutil.move(dir+'/'+j,XML+'/'+j)