Read xml files df

catchment/models.py

@@ -35,11 +35,38 @@ def read_variable_from_csv(filename):
 
     return newdataset
 
+
+def read_variable_from_xml(filename):
+    """Reads a named variable from a XML file, and returns a
+    pandas dataframe containing that variable. The XML file must contain
+    a column of dates, a column of site ID's, and (one or more) columns
+    of data - only one of which will be read.
+
+    :param filename: Filename of XML to load
+    :return: 2D array of given variable. Index will be dates,
+             Columns will be the individual sites
+    """
+    dataset = pd.read_xml(filename)
+    dataset = dataset.rename({'Date':'OldDate', "Site_Name": "Site Name", "Rainfall_mm": "Rainfall (mm)"}, axis='columns')
+    dataset['Date'] = [pd.to_datetime(x, dayfirst=True) for x in dataset['OldDate']]
+    dataset = dataset.drop('OldDate', axis='columns')
+
+    newdataset = pd.DataFrame(index=dataset['Date'].unique())
+
+    for site in dataset['Site Name'].unique():
+        newdataset[site] = dataset[dataset['Site'] == site].set_index('Date')["Rainfall (mm)"]
+
+    newdataset = newdataset.sort_index()
+
+    return newdataset
+
+
 def daily_total(data):
     """Calculate the daily total of a 2D data array.
     Index must be np.datetime64 compatible format."""
     return data.groupby(data.index.date).sum()
 
+
 def daily_mean(data):
     """Calculate the daily mean of a 2D data array.
     Index must be np.datetime64 compatible format."""