Sorting XML tags by child elements Python

时间:2016-10-20 12:39:23

标签: python python-2.7 lxml elementtree

I have a number of 'root' tags with children 'name'. I want to sort the 'root' blocks, ordered alphabetically by the 'name' element. Have tried lxml / etree / minidom but can't get it working... I can't get it to parse the value inside the tags, and then sort the parent root tags.

<?xml version='1.0' encoding='UTF-8'?>
  <roots>
    <root>
      <path>//1.1.1.100/Alex</path>
      <name>Alex Space</name>
    </root>
    <root>
      <path>//1.1.1.101/Steve</path>
      <name>Steve Space</name>
    </root>
    <root>
      <path>//1.1.1.150/Bethany</path>
      <name>Bethanys</name>
    </root>
</roots>

Here is what I have tried:

import xml.etree.ElementTree as ET

    def sortchildrenby(parent, child):
        parent[:] = sorted(parent, key=lambda child: child)


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

sortchildrenby(root, 'name')
for child in root:
    sortchildrenby(child, 'name')


tree.write('output.xml')

2 个答案:

答案 0 :(得分:2)

如果您想先放置名称节点:

x = """
  <roots>
    <root>
      <path>//1.1.1.100/Alex</path>
      <name>Alex Space</name>
    </root>
    <root>
      <path>//1.1.1.101/Steve</path>
          <name>Bethanys</name>
    </root>
    <root>
      <path>//1.1.1.150/Bethany</path>
        <name>Steve Space</name>
    </root>
</roots>"""

import lxml.etree as et
tree = et.fromstring(x)

for r in tree.iter("root"):
    r[:] = sorted(r, key=lambda ch: -(ch.tag == "name"))

print(et.tostring(tree).decode("utf-8"))

哪会给你:

<roots>
    <root>
      <name>Alex Space</name>
    <path>//1.1.1.100/Alex</path>
      </root>
    <root>
      <name>Bethanys</name>
    <path>//1.1.1.101/Steve</path>
          </root>
    <root>
      <name>Steve Space</name>
    <path>//1.1.1.150/Bethany</path>
        </root>
</roots>

但是如果你只想先添加它们就没有必要排序,你可以删除并重新插入名称到索引0:

import lxml.etree as et
tree = et.fromstring(x)

for r in tree.iter("root"):
    ch = r.find("name")
    r.remove(ch)
    r.insert(0, ch)

print(et.tostring(tree).decode("utf-8"))

如果节点实际上不是按排序顺序,并且您想按字母顺序重新排列 roots 节点:

x = """
  <roots>
    <root>
      <path>//1.1.1.100/Alex</path>
      <name>Alex Space</name>
    </root>
    <root>
      <path>//1.1.1.101/Steve</path>
         <name>Steve Space</name>
    </root>
    <root>
       <path>//1.1.1.150/Bethany</path>
       <name>Bethanys</name>
    </root>
</roots>"""
import lxml.etree as et
tree = et.fromstring(x)

tree[:] = sorted(tree, key=lambda ch: ch.xpath("name/text()"))

print(et.tostring(tree).decode("utf-8"))

哪会给你:

 <roots>
    <root>
      <path>//1.1.1.100/Alex</path>
      <name>Alex Space</name>
    </root>
    <root>
       <path>//1.1.1.150/Bethany</path>
       <name>Bethanys</name>
    </root>
    <root>
      <path>//1.1.1.101/Steve</path>
         <name>Steve Space</name>
    </root>
</roots>

您还可以结合前两种方法中的任何一种,重新排列 root 节点,将名称放在第一位。

答案 1 :(得分:-1)

试试这个:

import xml.etree.ElementTree as ET


xml="<?xml version='1.0' encoding='UTF-8'?><roots><root><path>//1.1.1.100/Alex</path><name>Alex Space</name></root><root><path>//1.1.1.101/Steve</path><name>Steve Space</name></root><root><path>//1.1.1.150/Bethany</path><name>Bethanys</name></root></roots>"
oldxml = ET.fromstring(xml)

names = []
for rootobj in oldxml.findall('root'):
    names.append(rootobj.find('name').text)

newxml = ET.Element('roots')
for name in sorted(names):
    for rootobj in oldxml.findall('root'):
        if name == rootobj.find('name').text:
            newxml.append(rootobj)
ET.dump(oldxml)
ET.dump(newxml)

我从变量中读取并将其转储到屏幕上。

您可以将其从文件中读取并将其转储到您需要的文件中。