PE plots now correctly display initial parameter distribution

graphics.py

@@ -133,6 +133,7 @@ def plot_CBL_PE(exp,figtitle,parameter_id=0,plot_spread=False,ax=None):
         parameter_number = exp.nr.parameter_number
         par_tran = exp.da.backgrounds[:,-parameter_number:,:][:,parameter_id,:]
         par_phys = pt[parameter_id](par_tran,kind='inv')
+        initpar_phys = pt[parameter_id](exp.da.initial_perturbed_parameters,kind='inv')
         t = exp.da.time/3600.
 
         plot_parameter_histogram = True
@@ -187,7 +188,18 @@ def plot_CBL_PE(exp,figtitle,parameter_id=0,plot_spread=False,ax=None):
         else:
             ax1 = ax
 
-        ax1.step(t,par_phys.mean(axis=1),color=mean_color)
+        # Initial parameter distribution
+        ax1.step([0,t[0]],np.median(initpar_phys,axis=1)*np.array([1,1]),color=mean_color)
+        bmin = np.percentile(initpar_phys, 10)
+        bmax = np.percentile(initpar_phys, 90)
+        ax1.fill_between([0,t[0]],\
+                    y1=[bmin,bmin],\
+                    y2=[bmax,bmax],\
+                    color=spread_color,
+                    edgecolor=None,
+                    alpha=0.3)
+        # Later times
+        ax1.step(t,np.median(par_phys,axis=1),color=mean_color)
         for i in range(t.size-1):
             bmin = np.percentile(par_phys, 10, axis=1)[i+1]
             bmax = np.percentile(par_phys, 90, axis=1)[i+1]