天天看點

超簡單的主成分分析(PCA)協方差矩陣算法及其相關性分析熱力圖

相信了解過PCA的小夥伴都知道怎麼求協方差矩陣的公式

下面直接給出求相關性矩陣計算代碼及熱力圖

import seaborn as sns;sns.set(color_codes= True)

# 相關性分析
def covriance(data):    # 傳入的資料集-data
    X = d.corr()
    print(X)
    sns.heatmap(X,square=True, annot=True)
    plt.title("Correlation matrix")
    plt.show()
if __name__ == "__main__":
    covriance(dataset)      

相關性熱力圖效果如下

超簡單的主成分分析(PCA)協方差矩陣算法及其相關性分析熱力圖

繼續閱讀