Python数据相关系数矩阵和热力图轻松实现_python dataframe数据相关性生成相关矩阵图

对其中的参数进行解释

• 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 snsimport matplotlib.pyplot as pltdef 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()

  
  
1
2
3
4
5
6
7
8
1