pandas计算相关性并画热力图
实现这个功能有很多方法,但是下面的方法还是比较优雅的:
cols = ["ASSET", "HOUSES", "INCOME", "DEBT", "EDUC"]
corr = df[cols].corr()
corr.style.background_gradient(axis=None)
讲解:
corr.style:将相关性矩阵 corr 转换为 Styler 对象,允许对 DataFrame 进行样式化操作。
.background_gradient(axis=None):使用背景颜色渐变来可视化相关性矩阵。axis=None
表示对整个矩阵应用渐变,而不是按行或列应用。