1. 程式人生 > >Python資料相關係數矩陣和熱力圖輕鬆實現

Python資料相關係數矩陣和熱力圖輕鬆實現

對其中的引數進行解釋
  • plt.subplots(figsize=(9, 9))設定畫面大小,會使得整個畫面等比例放大的
  • sns.heapmap()這個當然是用來生成熱力圖的啦
  • df是DataFrame, pandas的這個類還是很常用的啦~
  • df.corr()就是得到這個dataframe的相關係數矩陣
  • 把這個矩陣直接丟給sns.heapmap中做引數就好啦
  • sns.heapmap中annot=True,意思是顯式熱力圖上的數值大小。
  • sns.heapmap中square=True,意思是將圖變成一個正方形,預設是一個矩形
  • sns.heapmap中cmap="Blues"是一種模式,就是圖顏色配置方案啦,我很喜歡這一款的。
  • sns.heapmap中vmax是顯示最大值
import seaborn as sns
import matplotlib.pyplot as plt
def test(df):
    dfData = df.corr()
    plt.subplots(figsize=(9, 9)) # 設定畫面大小
    sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues")
    plt.savefig('./BluesStateRelation.png')
    plt.show()