原始

首先准备一个文件,我用的是csv文件
我们先看一下最原始的效果:
在这里插入图片描述
代码为:

student = pd.read_csv('./input/small_data.csv')
plt.figure(figsize=(25, 10))
sns.heatmap(student.corr(), annot=True, cmap='YlGnBu')
plt.savefig('relevance.png')
plt.show()

可以看到每个字都很小,那么接下来开始我们的调整

图内

调整图内,我们只需要在heatmap的属性中添加一个这个属性:

annot_kws={"fontsize":20}

调整后的代码如下:

student = pd.read_csv('./input/small_data.csv')
plt.figure(figsize=(25, 10))
sns.heatmap(student.corr(), annot=True, cmap='YlGnBu', annot_kws={"fontsize":20})
plt.savefig('relevance.png')
plt.show()

效果如下:
在这里插入图片描述
可以看到,我们图内的字体已经变大了。

坐标轴

基于我们刚才的结果,坐标轴的信息只需要添加这两行代码,分别调整x轴和y轴:

plt.xticks(fontsize=20)
plt.yticks(fontsize=20)

调整后的代码如下:

student = pd.read_csv('./input/small_data.csv')
plt.figure(figsize=(25, 10))
sns.heatmap(student.corr(), annot=True, cmap='YlGnBu', annot_kws={"fontsize":20})
plt.xticks(fontsize=20)
plt.yticks(fontsize=20)
plt.savefig('relevance.png')
plt.show()

效果如下:
在这里插入图片描述
坐标轴的字体也变大了

color bar

最后我们来调整color bar的大小
首先我们要获取图像的句柄ax

ax = sns.heatmap(student.corr(), annot=True, cmap='YlGnBu', annot_kws={"fontsize":20})

然后调整color bar

cbar = ax.collections[0].colorbar
cbar.ax.tick_params(labelsize=20)

调整后的代码如下:

student = pd.read_csv('./input/small_data.csv')
plt.figure(figsize=(25, 10))
ax = sns.heatmap(student.corr(), annot=True, cmap='YlGnBu', annot_kws={"fontsize":20})
cbar = ax.collections[0].colorbar
cbar.ax.tick_params(labelsize=20)
plt.xticks(fontsize=20)
plt.yticks(fontsize=20)
plt.savefig('relevance.png')

效果如下:
在这里插入图片描述
至此,我们完成热力图的图像调整

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐