# 计算各变量之间的相关系数 corr = data.corr() # 生成热力图 sns.heatmap(corr, cmap='coolwarm') plt.show()
时间: 2024-03-19 15:40:55 浏览: 125
计算相关系数并绘制热力图
这是一个使用seaborn库在Python中生成热力图的代码示例。首先,使用pandas库中的corr函数计算数据集中各变量之间的相关系数,结果保存在变量'corr'中。然后,使用seaborn库中的heatmap函数生成热力图。将'corr'作为参数传递给heatmap函数,cmap参数用于设置颜色映射。在此示例中,使用了'coolwarm'颜色映射。最后,使用show函数显示图表。热力图可以直观地展示各变量之间的相关性,有助于发现数据集中的特征。
阅读全文