Python如何根据CSV排序对Xml文件进行排序？

import csv
import xml.etree.ElementTree as ET

sort_dict = {}
with open('data.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        sort_dict[row[0]] = int(row[1])

tree = ET.parse('data.xml')
root = tree.getroot()

sorted_elements = sorted(root, key=lambda elem: sort_dict.get(elem.tag, float('inf')))

root[:] = sorted_elements

tree.write('sorted_data.xml')