triangulation.py 文件源码

def plot_dendro(fn, iZ):
    """Plot dendrogram"""
    plt.figure(figsize=(25, 10))
    plt.title('Hierarchical Clustering Dendrogram')
    plt.xlabel('sample index')
    plt.ylabel('distance')
    sch.dendrogram(iZ, leaf_rotation=90., leaf_font_size=8.)
    outfn = "%s.png"%fn
    if type(fn) is str and len(fn.split('.')[-1])==3:
        outfn = fn
    plt.savefig(outfn)