Corrected problems when pdf was a delta

fbcotter · Jun 7, 2018 · 483ee3b · 483ee3b
1 parent 7c413a1
commit 483ee3b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/plotters.py b/plotters.py
@@ -622,7 +622,15 @@ def plot_dtcwt(yl, yh, fig=None, f=np.abs, top=1, fmt='chw', imshow_kwargs={}):
         else:
             x = np.linspace(vmin, vmax, 50)
         # Fit a kernel to it
-        density = stats.gaussian_kde(yh_disp[j].ravel())
+        try:
+            density = stats.gaussian_kde(yh_disp[j].ravel())
+        except np.linalg.LinAlgError:
+            def delta(x):
+                if x == 0:
+                    return 1
+                else:
+                    return 0
+            density = np.vectorize(delta)
         y = density(x)
         # Plot it vertically and then make it right-aligned
         hist.set_xlim(y.max(), 0)