两个读写 XML 文件的小函数

在编程的世界中，处理 XML 文件是一项常见的任务。本文将为您介绍两个实用的小函数，用于读取和写入 XML 文件，帮助您更高效地处理数据。

让我们来看看读取 XML 文件的函数。这个函数的主要目的是将 XML 文件中的数据解析并提取出来，以便在程序中进行进一步的处理和操作。

import xml.etree.ElementTree as ET

def read_xml_file(file_path):
    try:
        tree = ET.parse(file_path)
        root = tree.getroot()
        data = {}
        for child in root:
            key = child.tag
            value = child.text
            data[key] = value
        return data
    except Exception as e:
        print(f"读取 XML 文件时发生错误: {e}")
        return None

上述函数使用了 Python 标准库中的 xml.etree.ElementTree 模块。通过 parse 方法读取文件并获取根节点，然后遍历根节点的子节点，将标签作为键，文本内容作为值，存储在一个字典中并返回。

接下来是写入 XML 文件的函数。它接收一些数据，并将其按照 XML 的格式写入到指定的文件中。

def write_xml_file(file_path, data):
    try:
        root = ET.Element('root')
        for key, value in data.items():
            child = ET.SubElement(root, key)
            child.text = value
        tree = ET.ElementTree(root)
        tree.write(file_path, encoding='utf-8', xml_declaration=True)
    except Exception as e:
        print(f"写入 XML 文件时发生错误: {e}")

在这个函数中，首先创建了一个根节点 root ，然后根据传入的数据创建子节点，并设置子节点的文本内容。最后，使用 ElementTree 的 write 方法将构建好的 XML 结构写入到文件中。

这两个小函数在处理 XML 文件时能够提供很大的便利。无论是从 XML 文件中读取数据进行分析，还是将处理后的数据以 XML 格式保存，它们都能够高效地完成任务。

在实际应用中，您可以根据具体的需求灵活运用这两个函数。比如，在数据采集和转换的场景中，读取 XML 文件获取原始数据，经过处理后再使用写入函数保存结果。

这两个读写 XML 文件的小函数虽然简单，但在许多编程项目中都能发挥重要的作用，为您的数据处理工作提供有力的支持。