Plotting function

plots/plot_error_radius.py

@@ -9,7 +9,6 @@
 import numpy as np
 import matplotlib.pyplot as plt
 
-
 def read_numeric(dataFilePath):
     import csv
 
@@ -27,30 +26,40 @@ def read_numeric(dataFilePath):
                 else:
                     data = np.append(data, row, axis=0)
             except ValueError:
-                print('skip line' + row[1])
+                print('skip line')
     return data
 
 ## set properties
 radius = [100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1100, 1200]
 network = 'm2m_lstm'
 # feature = ['air_temperature', 'humidity']
 # feature = ['air_temperature']
-features =  [['air_temperature'], ['air_temperature', 'wind_north', 'wind_east']]
+features =  [['air_temperature'], ['air_temperature', 'wind_north', 'wind_east'], ['air_temperature', 'humidity']]
 
 # set path
 path = os.path.join('..', 'out')
 dataFileID = 'log.csv'
 dataStatFileId = 'data_stat.csv'
 
+radiusS = [100, 200, 300, 400, 500, 600, 700, 800, 900, 1000]
+radiusL = [100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1100, 1200]
+
 # init some stuff for looping
-RMS = [[0]  * len(radius), [0]  * len(radius)]
-RMS_norm = [[0]  * len(radius), [0]  * len(radius)]
+RMS = [[0]  * len(radiusL), [0]  * len(radiusL), [0]  * len(radiusS)]
+RMSnorm = [[0]  * len(radiusL), [0]  * len(radiusL), [0]  * len(radiusS)]
 
+RMSTrain = [[0]  * len(radiusL), [0]  * len(radiusL), [0]  * len(radiusS)]
+RMSnormTrain = [[0]  * len(radiusL), [0]  * len(radiusL), [0]  * len(radiusS)]
 
 featCounter = 0
 
 for feature in features:
 
+    if featCounter == 2:
+        radius = radiusS
+    else:
+        radius = radiusL
+
     rCounter = 0
     # loop over radii
     for r in radius:
@@ -65,26 +74,52 @@ def read_numeric(dataFilePath):
 
         print( data[-1, -1])
 
-        RMS_norm[featCounter][rCounter] = data[-1, -1]
+        RMSnorm[featCounter][rCounter] = data[-1, -1]
         RMS[featCounter][rCounter] = dataStat[0][1] * dataStat[0][1] * data[-1, -1]
-
+
+        RMSnormTrain[featCounter][rCounter] = data[-1, 1]
+        RMSTrain[featCounter][rCounter] = dataStat[0][1] * dataStat[0][1] * data[-1, 1]
+
         rCounter = rCounter + 1
 
     featCounter = featCounter + 1
-# plot results
-fig, ax = plt.subplots( nrows=1, ncols=1 ) 
-tempPlt, = ax.plot(radius, RMS[0], label='temperature')
-windPlt, = ax.plot(radius, RMS[1], label='temperatur, wind speed')
-plt.ylabel('Mean squared error [$^\circ$ C$^2$]')
-plt.xlabel('radius [km]')
+
+
+csfont = {'fontname':'Century Gothic'}
+
+valStyle = '-'
+trainStyle = '--'
+
+r = [245/255, 159/255,35/2555] # given by Marc
+b = [102/255, 161/255, 211/255] # given by Marc
+
+# plot results   
+fig, ax = plt.subplots( nrows=1, ncols=1, figsize=(10, 6) ) 
+plt.subplots_adjust(top=0.82)
+
+
+tempPlt, = ax.plot(radiusL, RMS[0], label='temperature, validate',  color=b, linestyle = valStyle)
+windPlt, = ax.plot(radiusL, RMS[1], label='temperature + wind speed, validate',  color=r, linestyle = valStyle)
+humidPlt, = ax.plot(radiusS, RMS[2], label='temperature + humidity, validate',  color='g', linestyle = valStyle)
+
+tempPltTrn, = ax.plot(radiusL, RMSTrain[0], label='temperature, train' ,color=b, linestyle = trainStyle)
+windPltTrn, = ax.plot(radiusL, RMSTrain[1], label='temperature + wind speed, train', color=r, linestyle = trainStyle)
+humidPltTrn, = ax.plot(radiusS, RMSTrain[2], label='temperature + humidity, train', color='g', linestyle = trainStyle)
+
+plt.ylabel('Mean squared error [$^\circ$C$^2$]', fontsize=16, **csfont)
+plt.xlabel('radius [km]', fontsize=16, **csfont)
 plt.grid(True)
-plt.title('Error for Different Features and Radii')
+plt.title('Error for Different Features and Radii', fontsize=20, **csfont , y=1)
+
+ax.set_xticklabels(ax.get_xticks(), fontsize=12, **csfont)
+ax.set_yticklabels(ax.get_yticks(), fontsize=12, **csfont)
 
 handles, labels = ax.get_legend_handles_labels()
-plt.legend(handles=[tempPlt, windPlt])
+plt.legend(handles=[tempPlt, tempPltTrn, windPlt, windPltTrn, humidPlt, humidPltTrn], 
+           fontsize=16, prop={'family': 'Century Gothic'}, 
+           bbox_to_anchor=(0., 1.02, 1., .102), loc=3, ncol=3, mode="expand", borderaxespad=0.)
 
 fig.savefig(os.path.join('..', 'fig', 'error_' + '_'.join(feature) + '.png'))
 fig.savefig(os.path.join('..', 'fig', 'error_' + '_'.join(feature) + '.eps'))
 
-plt.show(fig) 
-# plt.close(fig) 
+plt.show(fig)