Получить все дети из тега XML с помощью ElementTree
Я пытаюсь проанализировать XML-файл, используя ElementTree, и в какой-то момент я получаю только первый дочерний элемент вместо всех дочерних элементов внутри тега. Ниже приведена моя XML-структура:-
<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
<sentences>
<sentence id="2339">
<text>I charge it at night and skip taking the cord with me because of the good battery life.</text>
<aspectTerms>
<aspectTerm term="cord" polarity="neutral" from="41" to="45"/>
<aspectTerm term="battery life" polarity="positive" from="74" to="86"/>
</aspectTerms>
</sentence>
<sentence id="812">
<text>I bought a HP Pavilion DV4-1222nr laptop and have had so many problems with the computer.</text>
</sentence>
<sentence id="1316">
<text>The tech guy then said the service center does not do 1-to-1 exchange and I have to direct my concern to the "sales" team, which is the retail shop which I bought my netbook from.</text>
<aspectTerms>
<aspectTerm term="service center" polarity="negative" from="27" to="41"/>
<aspectTerm term=""sales" team" polarity="negative" from="109" to="121"/>
<aspectTerm term="tech guy" polarity="neutral" from="4" to="12"/>
</aspectTerms>
</sentence>
</sentences>
Я хочу получить термин в каждом теге aspectTerm. Вот мой код для этого:
import xml.etree.ElementTree as ET
tree = ET.parse('Laptops_Train.xml')
root = tree.getroot()
df = pd.DataFrame()
def getAspect(sentences):
reviewList = []
text = sentence.find('text').text
reviewList.append(text)
for aspectTerms in sentence.iter('aspectTerms'):
#for aspectTerm in aspectTerms.iter('aspectTerm'):
aspect = aspectTerms.find('aspectTerm').get('term')
print(aspect)
return aspect
aspectList = []
for sentences in root.iter('sentences'):
for sentence in sentences.iter('sentence'):
aspectList.append(getAspect(sentence))
Фактические результаты:
cord
class 'NoneType'
service center
Ожидаемый результат:
[cord, battery life]
[]
[service center,"sales" team, tech guy]
заранее спасибо
2 ответа
Решение
Это гораздо проще сделать с помощью библиотеки lxml, которая имеет xpath.
>>> from lxml import etree
>>> tree = etree.parse('Laptops_Train.xml')
>>> for aspectTerms in tree.xpath('.//aspectTerms'):
... aspectTerms.xpath('aspectTerm/@term')
...
['cord', 'battery life']
['service center', '"sales" team', 'tech guy']
Обратите внимание, что все aspectTerms
иметь Term
имущество; нет пустых, которые привели бы к None
,
Редактировать, вдохновленный комментарием.
>>> from lxml import etree
>>> tree = etree.parse('Laptops_Train.xml')
>>> for sentence in tree.xpath('.//sentence'):
... sentence.xpath('.//aspectTerm/@term')
...
['cord', 'battery life']
[]
['service center', '"sales" team', 'tech guy']
Таким образом, решение было использовать.findall вместо.find. Потому что.findall выбирает всех детей. Мое решение заключается в следующем:
def getAspect(sentences):
aspectList = []
reviewList = []
text = sentence.find('text').text
reviewList.append(text)
for aspectTerms in sentence.iter('aspectTerms'):
#for aspectTerm in aspectTerms.iter('aspectTerm'):
aspect = aspectTerms.findall('aspectTerm')#.get('term')
for aspectElem in aspect:
aspects = aspectElem.get('term')
aspectList.append(aspects)
print(aspects)
return aspectList
aspectList = []
for sentences in root.iter('sentences'):
for sentence in sentences.iter('sentence'):
aspectList.append(getAspect(sentence))