python - 如何使用 lxml.etree 在 XML 中查找相同的属性值？

Question

我有一个类似于下面输入的 XML。

这里我要聚合Sales Diesel 的值燃料类型。

如何遍历所有 <Tank>元素并阅读 fuelItem属性找到多次出现的相同燃料类型，然后总结 Sales属性值？

输入:

 <EnterpriseDocument>
      <FuelTankList>
        <Tank fuelItem="Petrol" Sales="1000" />
        <Tank  fuelItem="Diesel" Sales="2000" />
        <Tank  fuelItem="Diesel" Sales="3000" />
      </FuelTankList>
    </EnterpriseDocument>

首选输出:

<EnterpriseDocument>
  <FuelTankList>
    <Tank fuelItem="Petrol" netSalesQty="1000" />
    <Tank  fuelItem="Diesel" netSalesQty="5000" />
  </FuelTankList>
</EnterpriseDocument>

Answer 1

由于您使用的是 lxml，因此可以使用 XSLT 和 Muenchian Grouping按 fuelItem 属性对 Tank 元素进行分组。

例子...

XML 输入 (input.xml)

<EnterpriseDocument>
    <FuelTankList>
        <Tank fuelItem="Petrol" Sales="1000" />
        <Tank  fuelItem="Diesel" Sales="2000" />
        <Tank  fuelItem="Diesel" Sales="3000" />
    </FuelTankList>
</EnterpriseDocument>

XSLT 1.0 (测试.xsl)

<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
  <xsl:output indent="yes"/>
  <xsl:strip-space elements="*"/>

  <xsl:key name="tanks" match="Tank" use="@fuelItem"/>

  <xsl:template match="@*|node()">
    <xsl:copy>
      <xsl:apply-templates select="@*|node()"/>
    </xsl:copy>
  </xsl:template>

  <xsl:template match="FuelTankList">
    <xsl:copy>
      <xsl:apply-templates select="@*"/>
      <xsl:for-each select="Tank[count(.|key('tanks',@fuelItem)[1])=1]">
        <xsl:copy>
          <xsl:apply-templates select="@*"/>
          <xsl:attribute name="Sales">
            <xsl:value-of select="sum(key('tanks',@fuelItem)/@Sales)"/>
          </xsl:attribute>
        </xsl:copy>
      </xsl:for-each>
    </xsl:copy>
  </xsl:template>

</xsl:stylesheet>

python

from lxml import etree

tree = etree.parse("input.xml")
xslt = etree.parse("test.xsl")

new_tree = tree.xslt(xslt)

print(etree.tostring(new_tree, pretty_print=True).decode("utf-8"))

输出(标准输出)

<EnterpriseDocument>
  <FuelTankList>
    <Tank fuelItem="Petrol" Sales="1000"/>
    <Tank fuelItem="Diesel" Sales="5000"/>
  </FuelTankList>
</EnterpriseDocument>