update event plots to plot even with nans

Author: TomDonoghue

specparam/plts/event.py

@@ -54,7 +54,7 @@ def plot_event_model(event_model, **plot_kwargs):
                                figsize=plot_kwargs.pop('figsize', [10, 4 + 4 * n_bands]))
     axes = cycle(axes)
 
-    xlim = [0, time_model.n_time_windows]
+    xlim = [0, event_model.n_time_windows]
 
     # 01: aperiodic params
     alabels = ['offset', 'knee', 'exponent'] if has_knee else ['offset', 'exponent']

specparam/plts/templates.py

@@ -289,7 +289,7 @@ def plot_params_over_time(times, params, labels=None, title=None, colors=None,
 
 
 @check_dependency(plt, 'matplotlib')
-def plot_param_over_time_yshade(times, param, average='mean', shade='std', scale=1.,
+def plot_param_over_time_yshade(times, param, average='nanmean', shade='nanstd', scale=1.,
                                 color=None, ax=None, **plot_kwargs):
     """Plot parameter over time with y-axis shading.
 

specparam/utils/data.py

@@ -15,11 +15,15 @@
 AVG_FUNCS = {
     'mean' : np.mean,
     'median' : np.median,
+    'nanmean' : np.nanmean,
+    'nanmedian' : np.nanmedian,
 }
 
 DISPERSION_FUNCS = {
     'var' : np.var,
+    'nanvar' : np.nanvar,
     'std' : np.std,
+    'nanstd' : np.nanstd,
     'sem' : sem,
 }