k-mer plots are prettier now :D

k_mer_design/make_k_mer_library_figures.py

@@ -54,10 +54,15 @@ def main():
         linkage = hc.linkage(sp.distance.squareform(final_df_dtw),
                              method=method)
         g = sns.clustermap(final_df_dtw, row_linkage=linkage,
-                           col_linkage=linkage)
+                           col_linkage=linkage, cbar_pos=(.1, .815, .05, .18),
+                           cbar_kws={'aspect': 15})
 
-        g.ax_col_dendrogram.set_title(f'{method.capitalize()} clustering of k-mers on DTW distance')
-        # g.fig.subplots_adjust(top=.9)
+        g.fig.suptitle(f'{method.capitalize()} clustering of '
+                       f'k-mers on DTW distance',
+                       fontsize='xx-large')
+        g.fig.tight_layout(pad=0.3)
+        # Prevent square colourbar
+        g.ax_cbar.set_aspect(4)
         g.savefig(args.out_dir / f'kmers_{method}_clustering.png')