參考鏈接: https://www.cnblogs.com/caiyishuai/p/11184166.html
熱力圖: 參考鏈接: https://blog.csdn.net/a19990412/article/details/79304944
1 import seaborn as sns 2 import matplotlib.pyplot as plt 3 def test(df): 4 dfData = df.corr() 5 plt.subplots(figsize=(9, 9)) # 設置畫面大小 6 sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues") 7 plt.savefig('./BluesStateRelation.png') 8 plt.show() 9 10 test(data)
plt.subplots(figsize=(9, 9)) 設置畫面大小，會使得整個畫面等比例放大的
sns.heapmap() 用來生成熱力圖
df是DataFrame格式數據集
df.corr() 得到這個dataframe的相關系數矩陣
可把這個矩陣直接丟給sns.heapmap中做參數
sns.heapmap中annot=True，顯式熱力圖上的數值大小。
sns.heapmap中square=True，意思是將圖變成一個正方形，默認是一個矩形
sns.heapmap中cmap="Blues"是一種模式，就是圖顏色配置方案
sns.heapmap中vmax是顯示最大值