cm = df.columns.tolist()
xcorr = df[colnm].corr()
#设置右上三角不绘制
#mask为 和相关系数矩阵xcorr一样大的 全0(False)矩阵
mask = np.zeros_like(xcorr, dtype=np.bool)
# 将mask右上三角(列号》=行号)设置为True
mask[np.triu_indices_from(mask)] = True
#cmap是设置热图的颜色
cmap = sns.diverging_palette(220, 10, as_cmap=True)
#绘制热图
g = sns.heatmap(xcorr, mask=mask, cmap=cmap, square=True, annot=True, fmt='0.2f')
# xcorr : 数据矩阵
# mask : 为True的元素对应位置不会画出来(mask面具的意义)
# cmap: 颜色设置
# square: (True)代表行列长度一致,且绘制的每个也是方格
# annot : 在格内显示数据
# fmt :数据格式
热图绘制结果:
来源(天池-红酒数据质量分析):
https://tianchi.aliyun.com/notebook-ai/detail?spm=5176.12282042.0.0.db7620428pXjaF&postId=4662