利用Python轻松处理XML数据

在当今数据驱动的世界中，XML（可扩展标记语言）作为一种常用的数据交换格式，广泛应用于各种领域。Python以其简洁明了的语法和强大的库支持，成为了处理XML数据的理想工具。本文将带你一步步了解如何利用Python进行简单的XML数据处理。

利用Python轻松处理XML数据

安装必要的库

首先，我们需要安装一个用于处理XML的Python库。xml.etree.ElementTree是Python标准库中的一个模块，专门用于处理XML数据。由于它是标准库的一部分，因此无需额外安装，直接导入即可。

import xml.etree.ElementTree as ET

读取XML文件

假设我们有一个名为data.xml的XML文件，内容如下：

<data>
    <item>
        <name>Item1</name>
        <price>10.99</price>
    </item>
    <item>
        <name>Item2</name>
        <price>15.99</price>
    </item>
</data>

我们可以使用ET.parse方法来读取这个文件：

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

这里，root变量将存储XML文档的根元素。

遍历XML元素

接下来，我们可以遍历XML中的所有元素。例如，打印出每个item的name和price：

for item in root.findall('item'):
    name = item.find('name').text
    price = item.find('price').text
    print(f'Name: {name}, Price: {price}')

输出将是：

Name: Item1, Price: 10.99
Name: Item2, Price: 15.99

修改XML数据

有时，我们可能需要修改XML数据。例如，将所有item的价格提高10%：

for item in root.findall('item'):
    price = item.find('price')
    new_price = float(price.text) * 1.1
    price.text = str(new_price)

tree.write('updated_data.xml')

这样，新的价格将被写入到updated_data.xml文件中。

创建新的XML元素

我们还可以在XML中添加新的元素。例如，为每个item添加一个category元素：

for item in root.findall('item'):
    category = ET.SubElement(item, 'category')
    category.text = 'Electronics'

tree.write('new_data.xml')

新的XML文件new_data.xml将包含新增的category元素。

处理命名空间

在实际应用中，XML文件可能包含命名空间。处理命名空间时，我们需要使用nsmap（命名空间映射）：

xml_data = '''
<ns:data xmlns:ns="http://example.com">
    <ns:item>
        <ns:name>Item1</ns:name>
        <ns:price>10.99</ns:price>
    </ns:item>
</ns:data>
'''

root = ET.fromstring(xml_data)
ns = {'ns': 'http://example.com'}

for item in root.findall('ns:item', ns):
    name = item.find('ns:name', ns).text
    price = item.find('ns:price', ns).text
    print(f'Name: {name}, Price: {price}')