34 lines
906 B
Python
34 lines
906 B
Python
from defusedxml import ElementTree
|
|
from collections import deque
|
|
import matplotlib.pyplot as plt
|
|
|
|
data = {"Germany": deque()
|
|
, "France": deque()
|
|
, "Italy": deque()
|
|
, "United States": deque()
|
|
}
|
|
|
|
with open("data/API_NY.GDP.MKTP.KN_DS2_en_xml_v2_10230884.xml") as fin:
|
|
tree = ElementTree.parse(fin)
|
|
for record in tree.getroot().find("data").findall("record"):
|
|
this_data = {field.get("name"): field.text for field in record.findall("field")}
|
|
if(this_data["Country or Area"] in data):
|
|
if(this_data["Value"] != None):
|
|
data[this_data["Country or Area"]].append((this_data["Year"], this_data["Value"]))
|
|
|
|
|
|
|
|
class Data(object):
|
|
def __init__(self, raw_data):
|
|
self.x = [int(k) for k, v in raw_data]
|
|
self.y = [float(v) for k, v in raw_data]
|
|
|
|
data = {k: Data(v) for k,v in data.items()}
|
|
|
|
plots = [plt.plot(v.x, v.y, label=k)[0] for k, v in data.items()]
|
|
|
|
plt.legend(handles = plots)
|
|
|
|
plt.show()
|
|
|