python:数据可视化 相关系数热力图绘制
仅十余行代码可实现可视化的相关系数矩阵热力图
把相关系数的展示做成热力图,展示更为直观并且方便与他人交流
主要用到的是seaborn库里的heatmap()函数
1.准备相关包
import pandas as pd
import numpy as np
from sklearn.tree import DecisionTreeClassifier
import seaborn as sns
import matplotlib.pyplot as plt
2.使用pandas读取数据,括号内为具体的文件路径,使用head()函数查看数据表头
data_train_set = pd.read_csv("C:/Users/Desktop/test/data_set.csv")
data_train_set.head()
3.【重点】计算相关系数
d = data_train_set.corr()
display(d)
具体的相关系数计算结果如下:
4.【展示】使用heatmap()函数绘制相关系数矩阵热力图
plt.subplots(figsize = (12,12))
sns.heatmap(d,annot = True,vmax = 1,square = True,cmap = "Reds")
plt.show()
最终的绘制效果如图
Bogir.Y: 请问博主想要输出这个图片和相关结果,应该怎么写代码?
CSDN-Ada助手: 恭喜您写了第5篇博客!标题中的“四大国际化咨询项目分享”听起来非常吸引人。您对国际化咨询项目的经验和见解一定是非常宝贵的。不过,如果我可以提一个谦虚的建议的话,希望您在接下来的创作中能够更加详细地分享每个项目的具体案例和实施过程,这样读者们将能够更好地理解和应用您的经验。期待您未来更多精彩的创作!
周而易: 博主,怎么既显示相关性系数又显示显著性标记
Issac神经蛙: 使用的数据可以分享一下吗
鱼鱼爱干饭--: 另外如果指标有中文可以加一下这两行代码 plt.rcParams['font.sans-serif']=['SimHei'] plt.rcParams['axes.unicode_minus']=False