python怎么绘制相关性热力图
时间: 2023-09-24 21:13:47 浏览: 91
Python绘制热力图示例
4星 · 用户满意度95%
要绘制 Python 中的相关性热力图,您需要使用 Seaborn 库。您可以使用 Seaborn 中的 heatmap() 函数,该函数将 DataFrame 中的数据值绘制为颜色编码的矩形。以下是一个示例代码:
```python
import seaborn as sns
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5], 'B': [6, 7, 8, 9, 10], 'C': [11, 12, 13, 14, 15]})
# 计算相关性矩阵
corr = df.corr()
# 绘制相关性热力图
sns.heatmap(corr, annot=True, cmap='coolwarm')
```
该代码会生成一个矩阵,其中颜色编码反映了输入数据的值。如果值越高,则颜色越深,反之亦然。使用 annot=True,可以在每个矩形中显示相关系数的值。```cmap``` 参数可用于指定颜色图。
阅读全文