Python计算几个变量相关性 热力图代码
时间: 2023-03-03 15:36:36 浏览: 145
Python数据相关系数矩阵和热力图轻松实现教程
5星 · 资源好评率100%
以下是使用Python计算变量相关性并生成热力图的代码:
```python
import pandas as pd
import seaborn as sns
# 创建数据框
df = pd.DataFrame({'变量1': [1, 2, 3, 4, 5], '变量2': [2, 4, 6, 8, 10], '变量3': [1, 3, 5, 7, 9]})
# 计算相关系数
corr = df.corr()
# 生成热力图
sns.heatmap(corr, annot=True)
```
在这个例子中,我们创建了一个包含3个变量的数据框,并计算了这些变量之间的相关系数。然后,使用Seaborn库中的`heatmap`函数生成了热力图。`annot=True`参数用于在热力图中显示相关系数的数值。您可以将这段代码中的数据框替换为您自己的数据框来计算变量之间的相关性并生成热力图。
阅读全文