提示信息

使用Pearson相关系数与热力图进行相关性分析:附Python完整代码

扑朔迷离 已关注

使用Pearson相关系数进行相关性分析是统计分析中的一种常用方法。我们可以使用Python的pandasseaborn库来计算相关系数,并通过热力图进行可视化。以下是一个完整的代码示例:

  1. import pandas as pd
  2. import seaborn as sns
  3. import matplotlib.pyplot as plt
  4. # 创建一个示例数据集
  5. data = {
  6. 'A': [1, 2, 3, 4, 5],
  7. 'B': [5, 4, 3, 2, 1],
  8. 'C': [2, 3, 4, 5, 6],
  9. 'D': [5, 3, 6, 2, 4]
  10. }
  11. df = pd.DataFrame(data)
  12. # 计算Pearson相关系数
  13. corr_matrix = df.corr(method='pearson')
  14. # 使用Seaborn绘制热力图
  15. plt.figure(figsize=(8, 6))
  16. sns.heatmap(corr_matrix, annot=True, fmt=".2f", cmap='coolwarm', cbar=True)
  17. plt.title("Pearson Correlation Coefficient Heatmap")
  18. plt.show()

代码说明:

  1. 数据创建:首先,我们创建一个示例数据集,该数据集是一个字典,然后将其转换为DataFrame。

  2. 计算相关系数:使用pandascorr方法计算DataFrame中各列之间的Pearson相关系数。

  3. 绘制热力图:使用seaborn库中的heatmap函数绘制相关系数矩阵的热力图。参数annot=True用于在每个单元格中显示数值,fmt=".2f"控制数值格式,cmap参数选择了一个颜色映射方案 ('coolwarm'),cbar=True显示颜色条。

通过这个简单的代码,你可以轻松地分析和可视化数据中不同变量之间的线性关系。根据需要,你可以替换数据集或调整绘图参数以适应你的具体需求。

    遇到难题? "AI大模型GPT4.0、GPT" 是你的私人解答专家! 点击按钮去提问......
扑朔迷离 关注 已关注

最近一次登录:2024-11-20 22:40:39   

暂时还没有签名,请关注我或评论我的文章
×
免费图表工具,画流程图、架构图