python：数据可视化 相关系数热力图绘制

仅十余行代码可实现可视化的相关系数矩阵热力图

把相关系数的展示做成热力图，展示更为直观并且方便与他人交流

主要用到的是seaborn库里的heatmap()函数

1.准备相关包

import pandas as pd
import numpy as np
from sklearn.tree import DecisionTreeClassifier
import seaborn as sns
import matplotlib.pyplot as plt

           

2.使用pandas读取数据，括号内为具体的文件路径,使用head（）函数查看数据表头

data_train_set = pd.read_csv("C:/Users/Desktop/test/data_set.csv")
data_train_set.head()
           

3.【重点】计算相关系数

d = data_train_set.corr()
display(d)
           

  具体的相关系数计算结果如下：

4.【展示】使用heatmap()函数绘制相关系数矩阵热力图

plt.subplots(figsize = (12,12))
sns.heatmap(d,annot = True,vmax = 1,square = True,cmap = "Reds")
plt.show()
           

最终的绘制效果如图