diff --git a/solventmapcreator/clusteringanalysis/dendrogramplotting.py b/solventmapcreator/clusteringanalysis/dendrogramplotting.py index 438e776cc02809945aaaf2529184c182136ea2f1..75472062c1cafec53e77c251d9533bd89147c7c8 100644 --- a/solventmapcreator/clusteringanalysis/dendrogramplotting.py +++ b/solventmapcreator/clusteringanalysis/dendrogramplotting.py @@ -37,10 +37,13 @@ def create_dendogram(linkage_matrix, labels, figsize): """Creates a dendogram plot. """ dendrogram_plot, dendrogram_plot_axis = create_plot_with_axis(figsize) + threshold = 0.4*max(linkage_matrix[:,2]) cluster.dendrogram(linkage_matrix, labels=labels, orientation='right', leaf_rotation=0, leaf_font_size=7, + color_threshold=threshold, ax=dendrogram_plot_axis) append_plot_labels(dendrogram_plot_axis) + dendrogram_plot_axis.axvline(x=threshold, c='k') return dendrogram_plot, dendrogram_plot_axis def append_plot_labels(plot_axis):