分离文件夹下的xml文件和jpg文件

参考了: python分离不同后缀名的文件 - g7y12 - 博客园

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import os
import shutil

def deal_file(src,s1,s2):
    # 区分jpg和mp4
    xml = []
    jpg = []
    for f in os.listdir(src):
        if f.endswith('.xml'):
            xml.append(f)
        elif f.endswith('.jpg'):
            jpg.append(f)
    # 创建目标文件夹
    if not os.path.isdir(s1):
        os.mkdir(s1)
    if not os.path.isdir(s2):
        os.mkdir(s2)
    # 拷贝文件到目标文件夹
    for x in xml:
        xml_1 = os.path.join(src,x)
        shutil.copy(xml_1,s1)
    for j in jpg:
        jpg_2 = os.path.join(src,j)
        shutil.copy(jpg_2,s2)

if __name__ == "__main__":
     file = "./"
     src = os.path.join(file,'datafile')
     s1 = os.path.join(file, 'xml_file')
     s2 = os.path.join(file, 'jpg_file')
     deal_file(src, s1, s2)