本篇博文主要介绍如何用python画相关系数的热力图。

dataframe.corr()表示了dataframe中的两个变量之间的相关性,取值范围为[-1,1],取值接近-1,表示反相关,取值接近1,表正相关。

plt.rcParams['figure.figsize'] = (20, 15) 
sns.heatmap(diabetes_df.corr(), annot = True, linewidths=.5, cmap="YlGnBu") 
plt.title('Correlation between features', fontsize = 30)
plt.tight_layout()

Out:
在这里插入图片描述

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐