参考链接: 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是显示最大值