相关系数矩阵可视化(生成热力图)

对其中的参数进行解释
  • 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()
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值