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

会用到的库和方法:

  • 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()

你可能感兴趣的:(python绘图)