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 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
           

再分享一下我老师大神的人工智能教程吧。零基础!通俗易懂!风趣幽默!希望你也加入到我们人工智能的队伍中来!http://www.captainbed.net

这里写图片描述

你可能感兴趣的:(Python数据相关系数矩阵和热力图轻松实现)