Разобрать вложенный XML в DataFrame - PullRequest
Новая
       32

Разобрать вложенный XML в DataFrame

0 голосов
/ 15 октября 2019

Я получаю ответ от API в форме XML. Это выглядит так: 

<result_set>
    <status> ...
    </status>
    <copyright> ...
    </copyright>
    <results>
       <congress> ...
       </congress>
       <chamber> ...
       </chamber>
       <num_results> ...
       </num_results>
       <offset> ...
       </offset>
       <members>
         <member>
           <id> ... </id>
           <name> ... </name>
           ...
           ...
           ...
         </member>
         <member> ... </member>
         ...
         ... 
         ...
       </members>
    </results>
</result_set>

Мне нужны данные под каждым тегом в фрейме данных pandas вида: 

id name ... ... ...
1  JOHN ... ... ...
2  DOE  ... ... ...

Я пробовал ElementTree, но мне не удалось все попытки, поэтомудалеко.

1 Ответ

1 голос
/ 15 октября 2019

Ниже 

import pandas as pd
import xml.etree.ElementTree as ET

xml = '''<r> 
         <members>
         <member>
           <id>1</id>
           <name>jack</name>
         </member>
         <member>
           <id>5</id>
           <name>dan</name>
         </member>

       </members>
       </r>'''

root = ET.fromstring(xml)
data = []
members = root.findall('.//member')
for member in members:
    data.append({c.tag:c.text for c in list(member)})
df = pd.DataFrame(data)
print(df.head())

выход 

  id  name
0  1  jack
1  5   dan
...