python根据文件名来取出相同文件名其他格式的文件

分类好训练用的图片数据后划分相对应的标注文件

import  os,shutil

def  xml(fileDir, xmlDir):
    list2=os.listdir(fileDir)#从文件夹读取文件名到list2
    for  i  in list2:
        ext = os.path.splitext(i)
        j=ext[0]+'.xml'#改成标注文件的文件格式
        if os.path.exists(xmlDir+ '/' + j):
            shutil.move(xmlDir + j, new_xmlDir + "\" + j)#将原xml文件的路径移到新xml的文件路径
    return
if __name__ == '__main__':
    fileDir=r"D:4月任务4.14sleepperson1" + "\" #原图像路径
    xmlDir=r"D:4月任务4.14sleepval_xml" + "\"#原xml路径
    new_xmlDir=r'D:4月任务4.14sleepperson1' #新xml路径
    xml(fileDir, xmlDir)