fix in case sample size is too small to create density colors

erdogant · Jun 3, 2022 · 44daee1 · 44daee1
1 parent 0646046
commit 44daee1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scatterd/scatterd.py b/scatterd/scatterd.py
@@ -227,13 +227,14 @@ def gradient_on_density_color(X, c_rgb, labels):
             else:
                 xy = np.vstack([X[idx, 0], X[idx, 1], X[idx, 2]])
 
-            if len(idx)>1:
+            try:
                 # Compute density
                 z = gaussian_kde(xy)(xy)
                 # Sort on density
                 didx = idx[np.argsort(z)[::-1]]
-            else:
+            except:
                 didx=idx
+
             # order colors correctly based Density
             density_colors[didx] = c_rgb[idx, :]
             # plt.figure()