Minor plot fixes

dpeerlab · Aug 22, 2019 · 73aa8fd · 73aa8fd
1 parent ad3c378
commit 73aa8fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/palantir/plot.py b/src/palantir/plot.py
@@ -258,8 +258,8 @@ def plot_gene_expression(data, tsne, genes, plot_scale=False,
     for g, ax in zip(genes, fig):
         # Data
         c = data.loc[cells, g]
-        vmin = np.percentile(c, percentile)
+        vmin = np.percentile(c[~np.isnan(c)], percentile)
-        vmax = np.percentile(c, 100 - percentile)
+        vmax = np.percentile(c[~np.isnan(c)], 100 - percentile)
 
         ax.scatter(tsne['x'], tsne['y'], s=3, color='lightgrey')
         ax.scatter(tsne.loc[cells, 'x'], tsne.loc[cells, 'y'], s=3,