Jpg format and XML format files are separated into different folders

import os
import shutil

root = "./data"
dataset ="./data/datasets"   # Files that need to be separated , namely image  And xml Under this file 

img = root + "/JPEGImages"
annotation = root + "/Annotations"

if not os.path.exists(img):
    os.mkdir(img)
if not os.path.exists(annotation):
    os.mkdir(annotation)

for x in os.scandir(dataset):
    if x.name.endswith(".jpg"):
        needpath = "./data/datasets/{name}".format(name=x.name)
        shutil.move(needpath,img)
    if x.name.endswith(".xml"):
        needpath ="./data/datasets/{name}".format(name=x.name)
        shutil.move(needpath,annotation)