Update plot.py

vmax = max(1, color_max)
sejooning · Oct 1, 2015 · d748b3d · d748b3d
1 parent 0af8ae4
commit d748b3d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyloh/postprocess/plot.py b/pyloh/postprocess/plot.py
@@ -63,8 +63,8 @@ def plot_BAF_heatmap_by_segment(BAF_counts, seg_name, outheatmap_dir_name):
     pyplot.yticks(scipy.linspace(0, 100, 11), scipy.linspace(0, 1, 11))
     pyplot.xlabel('Tumor sample B allele frequency')
     pyplot.ylabel('Normal sample B allele frequency')
-    pyplot.imshow(BAF_counts, vmin = 0, vmax = color_max)
+    pyplot.imshow(BAF_counts, vmin = 0, vmax = max(1, color_max))
     cbar = pyplot.colorbar(ticks=[0, color_max], orientation='vertical', shrink=0.78)
     cbar.ax.set_yticklabels(['0', '>= ' + str(int(color_max))])
     pyplot.savefig('./' + outheatmap_dir_name + '/' + seg_name, bbox_inches='tight')
-
+