怎么把上面的corr得到的矩阵画一个heatmap
时间: 2024-03-17 13:42:40 浏览: 19
要将 corr 得到的矩阵画成 heatmap,可以使用 Python 中的 seaborn 库。
给定一个 DataFrame,可以使用 DataFrame.corr() 方法计算各列之间的相关系数矩阵。然后,使用 seaborn 库的 heatmap() 函数将该矩阵画成 heatmap。
以下是一个示例代码:
``` python
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
corr_matrix = df.corr()
sns.heatmap(corr_matrix, annot=True, cmap='coolwarm')
plt.show()
```
在该示例中,首先创建了一个 DataFrame,其中包含三列。然后,使用 DataFrame.corr() 方法计算了列之间的相关系数矩阵。最后,使用 seaborn 库的 heatmap() 函数画出了该矩阵的 heatmap。
heatmap() 函数的 annot 参数用于控制是否在 heatmap 上标注数值。cmap 参数用于指定 color map,即颜色映射。在上面的代码中,使用了 'coolwarm' 颜色映射。
运行上述代码,就可以得到一个类似于下图的 heatmap:
![heatmap](https://img-blog.csdn.net/20180920211031757?watermark/2/text/aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2p1bGlhbmtp/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/q/85)
其中,颜色越浅表示相关性越高,颜色越深表示相关性越低。在该示例中,由于每列的值都是线性相关的,因此相关系数矩阵中的所有值都为 1,因此 heatmap 上的颜色都为浅蓝色。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)