如何将csv列表传递给XML findall并返回匹配的子节点

Question

在这里，我想将csv列表传递到我的python代码中，以仅检索在csv列表中匹配的Childone值。这样做的正确方法是什么？

import xml.etree.ElementTree as ET    
import csv        
tree = ET.parse("C:/Users/thisuser/Desktop/stack.xml")
root = tree.getroot()

with open('C:/Users/thisuser/Desktop/stack.csv', 'rb') as f:
    reader = csv.reader(f)
    your_list = list(reader)

for ticket in root.findall('.//Parent'):
    Childone = ticket.find('.//Childone').text
    for Childone in your_list:

        for ticket2 in ticket.findall('.//ParentTwo'):
            Childtwo = ticket2.find('.//Childtwo').text
            print "Childone={Childone}, Childtwo={Childtwo}".format(Childone=Childone, Childtwo=Childtwo)

期望的结果

ChildOne=2  ChildTwo=c
ChildOne=2  ChildTwo=d

ChildOne=3  ChildTwo=c
ChildOne=3  ChildTwo=e

目前的结果

Childone=['2'], Childtwo=a
Childone=['2'], Childtwo=b
Childone=['3'], Childtwo=a
Childone=['3'], Childtwo=b
Childone=['2'], Childtwo=c
Childone=['2'], Childtwo=d
Childone=['3'], Childtwo=c
Childone=['3'], Childtwo=d
Childone=['2'], Childtwo=c
Childone=['2'], Childtwo=e
Childone=['3'], Childtwo=c
Childone=['3'], Childtwo=e

XML SAMPLE

<Parent>
    <Childone>1</Childone>
        <ParentTwo>
            <Childtwo>a</Childtwo>
        </ParentTwo>
        <ParentTwo>
            <Childtwo>b</Childtwo>
        </ParentTwo>
</Parent>

<Parent>
    <Childone>2</Childone>
        <ParentTwo>
            <Childtwo>c</Childtwo>
        </ParentTwo>
        <ParentTwo>
            <Childtwo>d</Childtwo>
        </ParentTwo>
</Parent>

<Parent>
    <Childone>3</Childone>
        <ParentTwo>
            <Childtwo>c</Childtwo>
        </ParentTwo>
        <ParentTwo>
            <Childtwo>e</Childtwo>
        </ParentTwo>
</Parent>

EXCEL SAMPLE

我收到此错误：

AttributeError：'NoneType'对象没有属性'text'

Answer 1

你可以迭代你的csv并只处理具有匹配的Parent元素的Childone元素（使用XPath predicate）。

此外，您已经知道Childone元素值，因此您只需要Childtwo值...

XML输入（test.xml）

<doc>
    <Parent>
        <Childone>1</Childone>
        <ParentTwo>
            <Childtwo>a</Childtwo>
        </ParentTwo>
        <ParentTwo>
            <Childtwo>b</Childtwo>
        </ParentTwo>
    </Parent>

    <Parent>
        <Childone>2</Childone>
        <ParentTwo>
            <Childtwo>c</Childtwo>
        </ParentTwo>
        <ParentTwo>
            <Childtwo>d</Childtwo>
        </ParentTwo>
    </Parent>

    <Parent>
        <Childone>3</Childone>
        <ParentTwo>
            <Childtwo>c</Childtwo>
        </ParentTwo>
        <ParentTwo>
            <Childtwo>e</Childtwo>
        </ParentTwo>
    </Parent>
</doc>

CSV（test.csv）

2
3

蟒蛇

import xml.etree.ElementTree as ET
import csv

tree = ET.parse("test.xml")

with open('test.csv', 'rb') as f:
    reader = csv.reader(f)
    for i, row in enumerate(reader):
        if i:
            print ""
        for childtwo in tree.findall(".//Parent[Childone='{}']/ParentTwo/Childtwo".format(row[0])):
            print "ChildOne={childone}  ChildTwo={childtwo}".format(childone=row[0], childtwo=childtwo.text)

产量

ChildOne=2  ChildTwo=c
ChildOne=2  ChildTwo=d

ChildOne=3  ChildTwo=c
ChildOne=3  ChildTwo=e