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()
           

最終的繪制效果如圖