当前位置: 移动技术网 > IT编程>脚本编程>Python > python制图【1】相关系数热力图

python制图【1】相关系数热力图

2020年08月10日  | 移动技术网IT编程  | 我要评论
会用到的库和方法:pandas 中 corr()、read_csv()matplotlib中figure()seaborn中heatmap()import pandas as pdimport matplotlib.pyplot as pltimport seaborn as sns#导入文件file_path = r'G:\hotel_bookings.csv'df = pd.read_csv(file_path)#复制副本data_copy=df.copy(deep=Tru

会用到的库和方法:

  • pandas 中 corr()、read_csv()
  • matplotlib中figure()
  • seaborn中heatmap()
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

#导入文件
file_path = r'G:\hotel_bookings.csv'
df = pd.read_csv(file_path)

#复制副本
data_copy=df.copy(deep=True)
#计算相关系数
cor = data_copy.corr()
#设置画面
plt.figure(figsize=(37,37))
#作图,cmap为颜色类型,annot=False不显示数字,vmax最大值,vmin最小值,suqare=true正方形,linewidths划分每个单元格的行的宽度,xticklabels,yticklabels
sns.heatmap(cor,annot=False,vmax=1,square=True,cmap='PuBu',linewidths=.5)
plt.savefig('./cor.jpg')
plt.show()

本文地址:https://blog.csdn.net/IN_Joseph/article/details/107882706

如您对本文有疑问或者有任何想说的,请 点击进行留言回复,万千网友为您解惑!

相关文章:

验证码:
移动技术网