时间戳

首页 > 解决方案 > 合并输出循环 lxml

python - 合并输出循环 lxml

问题描述

我想从 xml 中取出一些从变量中查找的元素。

这是my.xml文件：

<?xml version='1.0' encoding='UTF-8'?>
<ArrayOfSalesOrderHeader xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
    <SalesOrderHeader>
        <TenantCode>15152343</TenantCode>
        <SalesOrderDetails>
            <SalesOrderDetail>
                <ItemCode>20072129</ItemCode>
            </SalesOrderDetail>
            <SalesOrderDetail>
                <ItemCode>67332054</ItemCode>
            </SalesOrderDetail>
            <SalesOrderDetail>
                <ItemCode>20206133</ItemCode>
            </SalesOrderDetail>
            <SalesOrderDetail>
                <ItemCode>62071796</ItemCode>
            </SalesOrderDetail>
        </SalesOrderDetails>
    </SalesOrderHeader>
</ArrayOfSalesOrderHeader>

这是我的脚本：

doc = ET.parse("my.xml")
arrDat = '20206133'
fol = doc.xpath('.//SalesOrderDetail[descendant::ItemCode[not(contains(text(),"' + arrDat + '"))]]')
for SOD in fol :
    SOD.getparent().remove(SOD)

doc.write('output.xml', xml_declaration=True, encoding='utf-8', method="xml")

我定义arrDat为数组时的问题：

doc = ET.parse("my.xml")
arrDat = ['20072129','67332054']
cnt = 0
while cnt < len(arrDat) :
    fol = doc.xpath('.//SalesOrderDetail[descendant::ItemCode[not(contains(text(),"' + arrDat[cnt] + '"))]]')
    for SOD in fol :
        SOD.getparent().remove(SOD)


    doc.write('output.xml', xml_declaration=True, encoding='utf-8', method="xml")
    cnt += 1

我需要 output.xml 是这样的：

<?xml version='1.0' encoding='UTF-8'?>
<ArrayOfSalesOrderHeader xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
    <SalesOrderHeader>
        <TenantCode>15152343</TenantCode>
        <SalesOrderDetails>
            <SalesOrderDetail>
                <ItemCode>20072129</ItemCode>
            </SalesOrderDetail>
            <SalesOrderDetail>
                <ItemCode>67332054</ItemCode>
            </SalesOrderDetail>
        </SalesOrderDetails>
    </SalesOrderHeader>
</ArrayOfSalesOrderHeader>

标签： pythonpython-3.xlxml

解决方案

我认为您可以简单地检查项目节点值并删除列表中不存在的那个。这是实现：

from lxml import etree as ET

doc = ET.parse("data1.xml")
arrDat = ['20072129', '67332054']

for order in doc.xpath("//SalesOrderDetail"):
    item = order.xpath('ItemCode')
    item_code = item[0].text
    if item_code not in arrDat:
        order.getparent().remove(order)

doc.write('output.xml', xml_declaration=True, encoding='utf-8', method="xml")

推荐阅读