Разбор XML дочерних тегов с тем же именем с помощью ElementTree - PullRequest
Новая
       61

Разбор XML дочерних тегов с тем же именем с помощью ElementTree

0 голосов
/ 24 апреля 2020

Я пытаюсь проанализировать файл XML (в Python, который является новым для меня) со следующей структурой: 

<xml>
<document>
<fit>
<grp> some tags </grp>
<prp> 
<p> <id> 1674 </id> </p>
<drp> 
<name> Joe </name>
<post> 
<company> abc </company>
<company> Ltd. </company>
</post>
</drp>
</prp>
</fit>
</document>
<document>
.
.
.
</xml>

Чтобы извлечь информацию, такую ​​как идентификатор, имя, компания и затем записать их Для csv я попробовал следующий код: 

tree = ET.parse(file)
root=tree.getroot()
with open(csvfile, 'a') as f:
    writer=csvDictWriter(f, ['ID', 'NAME', 'NCOMP'], delimiter=', ')
    writer.writeheader()
    result = {}
    for child in root.findall('./fit'):
        result['ID'] = ( "" .join(child.find('p').find('id').text))
        result['NAME'] = ( "" .join(child.find('drp').find('name'))
        result['NCOMP'] = ( "" .join(child.find('drp').find('post').find('company')
        writer.write(result)

Однако для названия компании я получаю только содержимое первого тега, затем я попытался использовать для l oop и добавить в список, подобный этому: 

Com = []
for each in child.find('drp').find('post'):
    coms = each.find('company')
    Com = Com.append[coms]
    result['NCOMP'] = Com

Желаемый вывод: 

ID.      NAME.     NCOMP
1674.    Joe.      abc Ltd.

Как изменить код, чтобы он включал значение обоих тегов?

Ответы [ 2 ]

0 голосов
/ 25 апреля 2020

Другое решение. 

from simplified_scrapy import SimplifiedDoc,req,utils
html = '''
<xml>
<document>
<fit>
<grp> some tags </grp>
<prp> 
<p> <id> 1674 </id> </p>
<drp> 
<name> Joe </name>
<post> 
<company> abc </company>
<company> Ltd. </company>
</post>
</drp>
</prp>
</fit>
</document>
<document>
.
.
.
</xml>
'''
doc = SimplifiedDoc(html)
rows = []
rows.append(['ID', 'NAME', 'NCOMP'])
for document in doc.documents:
  rows.append([document.id.text,document.name.text," ".join(document.companys.text)])
utils.save2csv('test.csv',rows)

Результат: 

ID,NAME,NCOMP
1674,Joe,abc Ltd.
0 голосов
/ 24 апреля 2020

Попробуйте что-нибудь в этом духе; он использует l xml для сбора данных через xpath и pandas для хранения их в фрейме данных: 

data = """
<xml>
   <document>
      <fit>
         <grp>some tags</grp>
         <prp>
            <p>
               <id>1674</id>
            </p>
            <drp>
               <name>Joe</name>
               <post>
                  <company>abc</company>
                  <company>Ltd.</company>
               </post>
            </drp>
         </prp>
      </fit>
   </document>
</xml>
"""

from lxml import etree
import pandas as pd

columns = ["ID", "NAME", "NCOMP"]
rows = []

doc = etree.XML(data)

targets = doc.xpath('//prp')
for target in targets:
    row = []
    id = target.xpath('./p/id/text()')[0]
    name = target.xpath('./drp/name/text()')[0]
    ncomp = target.xpath('./drp//post//company/text()')
    row.extend([id,name,' '.join(ncomp)])    
    rows.append(row)
pd.DataFrame(rows,columns=columns)

Вывод: 

    ID     NAME     NCOMP
0   1674    Joe     abc Ltd.

Правка - версия ET.

сначала: 

import xml.etree.ElementTree as ET

затем, начиная с doc, заменить на: 

doc = ET.fromstring(data)
et_targets = doc.findall('.//prp')
for target in et_targets:
    row = []
    id = target.findall('./p/id')[0]
    name = target.findall('./drp/name')[0]
    ncomp = target.findall('./drp//post//company')[0]
    row.extend([id.text,name.text,' '.join(ncomp.text)])    
    rows.append(row)
pd.DataFrame(rows,columns=columns)

Вывод должен быть таким же.

...