作者 | Julia Kho

編譯 | CDA數(shù)據(jù)分析師

Annotated Heatmaps of a Correlation Matrix in 5 Simple Steps

熱圖是數(shù)據(jù)的圖形表示，也就是說，它使用顏色來向讀者傳達價值。當您擁有大量數(shù)據(jù)時，這是一個很好的工具，可以幫助觀眾了解最重要的區(qū)域。

在本文中，我將指導(dǎo)您通過5個簡單步驟創(chuàng)建自己的相關(guān)矩陣注釋熱圖。

1. 導(dǎo)入數(shù)據(jù)
2. 創(chuàng)建關(guān)聯(lián)矩陣
3. 設(shè)置mask隱藏上三角
4. 在Seaborn中創(chuàng)建熱圖
5. 導(dǎo)出熱圖

1）導(dǎo)入數(shù)據(jù)

df = pd.read_csv("Highway1.csv"，index_col = 0)

該公路事故數(shù)據(jù)集包含汽車事故率，每百萬車輛英里的事故以及若干設(shè)計變量。

2）創(chuàng)建相關(guān)矩陣

corr_matrix = df.corr()

我們使用的是.corr 創(chuàng)建相關(guān)矩陣 。請注意，此矩陣中不存在htype列，因為它不是數(shù)字。我們需要使用htype來計算相關(guān)性。

df_dummy = pd.get_dummies(df.htype)

df = pd.concat([df，df_dummy]，axis = 1)

另外，請注意，相關(guān)矩陣的上三角部分與下三角對稱。因此，我們的熱圖不需要顯示整個矩陣。我們將在下一步隱藏上三角形。

3）設(shè)置mask隱藏上三角

mask = np.zeros_like(corr_matrix，dtype = np.bool)

mask[np.triu_indices_from(mask)] =True

讓我們打破上面的代碼吧。 np.zeros_like() 返回一個零數(shù)組，其形狀和類型與給定的數(shù)組相同。通過傳遞相關(guān)矩陣，我們得到如下的零數(shù)組。

該 dtype=np.bool 參數(shù)會覆蓋數(shù)據(jù)類型，因此我們的數(shù)組是一個布爾數(shù)組。

np.triu_indices_from(mask) 返回數(shù)組上三角形的索引。

現(xiàn)在，我們將上三角形設(shè)置為True。 mask[np.triu_indices_from(mask)]= True

現(xiàn)在，我們有一個掩碼可以用來生成熱圖。

4）在Seaborn中創(chuàng)建熱圖

f，ax = plt.subplots(figsize =(11,15))

heatmap=sns.heatmap(corr_matrix,

mask = mask,

square = True,

linewidths = .5,

cmap ='coolwarm',

cbar_kws = {'shrink':.4,

'ticks':[-1,-.5,0,0.5,1]},

vmin = -1,

vmax = 1,

annot = True,

annot_kws = {"size":12})

#增加列名做為標簽

ax.set_yticklabels(corr_matrix.columns，rotation = 0)

ax.set_xticklabels(corr_matrix.columns)

sns.set_style({'xtick.bottom':True},{'ytick.left':True})

為了創(chuàng)建我們的熱圖，我們傳遞步驟3中的相關(guān)矩陣和我們在步驟4中創(chuàng)建的蒙版以及自定義參數(shù)，以使我們的熱圖看起來更好。如果您有興趣了解每條線的作用，請參考以下參數(shù)說明。

#使每個單元格成方形

square = True,

#設(shè)置將每個單元格劃分為.5的行的寬度

linewidths = .5,

#Map數(shù)據(jù)值到coolwarm顏色空間

cmap ='coolwarm',

#Shrink在[-1，-.5,0,0.5,1]處的圖例大小和標簽刻度線

cbar_kws = {'shrink':.4,'ticks':[-1，-.5,0,0.5,1]},

#設(shè)置顏色條的最小值

vmin = -1,

#設(shè)置顏色條的最大值

vmax = 1,

#轉(zhuǎn)到相關(guān)值的注釋

annot = True,

#將注釋設(shè)置為12

annot_kws = {"size":12}

#將列名添加到x標簽

ax.set_xticklabels(corr_matrix.columns)

#將列名添加到y(tǒng)標簽并將文本旋轉(zhuǎn)到0度

ax.set_yticklabels(corr_matrix.columns,rotation = 0)

#在熱圖的底部和左側(cè)顯示標記

sns.set_style({'xtick.bottom':True},{'ytick.left':True})

5）導(dǎo)出熱圖 現(xiàn)在你有熱圖，讓我們把它導(dǎo)出。

heatmap.get_figure().savefig('heatmap.png', bbox_inches='tight')

如果您發(fā)現(xiàn)有一個非常大的熱圖不能正確導(dǎo)出，請使用bbox_inches = 'tight' 以防止圖像被切斷。