nan mean and nanstd

zm711 · Oct 4, 2023 · 5087ff6 · 5087ff6
1 parent f6b6666
commit 5087ff6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -812,8 +812,8 @@ def plot_response_trace(
                         )
             elif by_neuron:
                 for neuron in range(np.shape(response)[0]):
-                    avg_response = np.mean(response[neuron], axis=0)
-                    ebars = np.std(response[neuron], axis=0)
+                    avg_response = np.nanmean(response[neuron], axis=0)
+                    ebars = np.nanstd(response[neuron], axis=0)
                     if ebar:
                         self._plot_one_trace(
                             current_bins,
@@ -836,8 +836,8 @@ def plot_response_trace(
                         )
             elif by_trial:
                 for trial in range(np.shape(response)[1]):
-                    avg_response = np.mean(response[:, trial, :], axis=0)
-                    ebars = np.std(response[:, trial, :], axis=0)
+                    avg_response = np.nanmean(response[:, trial, :], axis=0)
+                    ebars = np.nanstd(response[:, trial, :], axis=0)
                     if ebar:
                         self._plot_one_trace(
                             current_bins,
@@ -859,8 +859,8 @@ def plot_response_trace(
                             stim_lines=current_length,
                         )
             else:
-                avg_response = np.mean(np.mean(response, axis=1), axis=0)
-                ebars = np.std(np.mean(response, axis=1), axis=0)
+                avg_response = np.nanmean(np.nanmean(response, axis=1), axis=0)
+                ebars = np.nanstd(np.nanmean(response, axis=1), axis=0)
                 if ebar:
                     self._plot_one_trace(
                         current_bins,