pd_multiple_analysis.py 文件源码

def save_heatmap(df, title, tag):
    """Create and save heatmaps."""
    sns.plt.figure()
    sns.plt.clf()
    filename = title+tag+".pdf"
    sns.heatmap(df, cmap="YlGn", annot=True, fmt='1.2f')
    sns.plt.title(title)
    sns.plt.ylabel(r'Number of samples $n$')
    sns.plt.xlabel(r'Number of dimensions $d$')
    sns.plt.savefig(filename)
    print("\t{} saved".format(filename))