sns_draw.py 文件源码

def corr_heatmap(df,cols=None,name=None):
    sns.set()
    if cols is None:
        cols = [i for i in df.columns.values if df[i].dtype!='object']
    df = df[cols].corr()
    print(df.shape)
    ds = sns.heatmap(df, annot=False)
    plt.show()
    if name is not None:
        ds.get_figure().savefig(name)