使用python将xml输入文件转换为所选元素csv输出文件

import xml.etree.ElementTree as ET

def extract_elements(xml_file, elements):
    tree = ET.parse(xml_file)
    root = tree.getroot()

    data = []
    for element in elements:
        elements = root.findall(element)
        for elem in elements:
            data.append(elem.text)

    return data

import csv

def write_to_csv(data, csv_file):
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(data)

import xml.etree.ElementTree as ET
import csv

def extract_elements(xml_file, elements):
    tree = ET.parse(xml_file)
    root = tree.getroot()

    data = []
    for element in elements:
        elements = root.findall(element)
        for elem in elements:
            data.append(elem.text)

    return data

def write_to_csv(data, csv_file):
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(data)

# 输入文件路径和所选元素列表
xml_file = 'input.xml'
elements = ['element1', 'element2', 'element3']

# 提取所选元素
data = extract_elements(xml_file, elements)

# 输出文件路径
csv_file = 'output.csv'

# 将数据写入CSV文件
write_to_csv(data, csv_file)