-
Notifications
You must be signed in to change notification settings - Fork 106
/
sframe_xml_to_dict.py
43 lines (32 loc) · 1.09 KB
/
sframe_xml_to_dict.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# You need the following package for this
# Installation : pip install xmldict
# Source : https://github.com/thoughtnirvana/xmldict
import xmldict
import graphlab as gl
def xml_filename_to_dict(filename):
"""
Reads the contents of an XML file, parses it and returns a dictionary.
Args:
filename: The name of the file with xml contents.
Returns:
A dictionary with the contents of the xml file.
>>> xml_filename_to_dict('file-1.xml')
{'foo': 'bar'}
"""
# Step 1: Read the contents of the file into a string
f = open(filename).readlines()
fstring = '\n'.join(f)
# Step 2: Convert the contents of the XML file into a dictionary
return xmldict.xml_to_dict(fstring)
# Let us assume that you have a set of files in the file "all_files.csv"
# Sample contents (of all_files.csv)
#
# file-1.xml
# file-2.xml
# ...
#
# file-200.xml
# Read the names of the files in to an SFrame
data = gl.SFrame.read_csv('all_files.csv', header=False)
data.rename({'X1': 'filename'})
data['xml-dict'] = data['filename'].apply(lambda x: xml_filename_to_dict(x))