python 解析xml文件

import os
import shutil
import xml.etree.ElementTree as ET

def xml_process(xml_save_path,xml_path):
    #xml_count = os.listdir(xml_path)
    os.chdir(xml_path)
    xml_count = os.listdir(xml_path)
    for xml_ in xml_count:
        tree = ET.parse(xml_)
        root = tree.getroot()
        #打印根节点标签名
        print('xml_root_name:'+root.tag)
        for object_ in root.findall('object'):
            name_ = object_.find('name').text
            if name_ == 'ship' or name_ == 'airplane':
                #如果标签中含有ship 或者是airplane 就将这个xml保存在新的文件夹中
                shutil.copy(xml_,xml_save_path)

if __name__ == "__main__":
    xml_path=r'D:\Postgraduate_study_materials\teacher\ahu_vot\dior_process\Annotations'
    xml_save_path = r'D:\Postgraduate_study_materials\teacher\ahu_vot\dior_process\new_save'
    xml_process(xml_save_path,xml_path)