绘制皮尔逊相关系数的热力图的代码
时间: 2024-12-14 15:07:36 浏览: 45
Python数据相关系数矩阵和热力图轻松实现教程
5星 · 资源好评率100%
绘制皮尔逊相关系数的热力图可以帮助我们直观地了解数据集中各变量之间的相关性。以下是使用Python和Seaborn库绘制皮尔逊相关系数热力图的代码示例:
```python
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
# 读取数据
data = pd.read_csv('your_data.csv')
# 计算皮尔逊相关系数
corr = data.corr(method='pearson')
# 绘制热力图
plt.figure(figsize=(10, 8))
sns.heatmap(corr, annot=True, cmap='coolwarm', fmt=".2f")
# 添加标题
plt.title('Pearson Correlation Heatmap')
# 显示图形
plt.show()
```
在这段代码中:
1. 使用`pandas`库读取数据文件。
2. 使用`corr`方法计算皮尔逊相关系数。
3. 使用`seaborn`库的`heatmap`函数绘制热力图,并设置`annot=True`以显示相关系数的具体数值,`cmap='coolwarm'`设置颜色映射,`fmt=".2f"`设置数值的格式。
阅读全文