天天看點

【相關系數熱力圖】相關系數矩陣可視化代碼模版

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

# 計算數值型變量間的相關性
corr = DF.corr()
# 相關性可視化
plt.figure(figsize = (15,15))
mask = np.zeros_like(corr)
mask[np.triu_indices_from(mask)] = True  # mask參數用來設定階梯圖
sns.heatmap(corr,annot=True, mask=mask, cmap="YlGnBu")
plt.show()
           

最終效果(效果示意圖,顔色不是上述代碼的指定顔色):

【相關系數熱力圖】相關系數矩陣可視化代碼模版