Python資料相關係數矩陣和熱力圖輕鬆實現
阿新 • • 發佈:2018-12-31
對其中的引數進行解釋
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()