注意一点,定义好了sns.heatmap()格式后,再利用ax.set_xticklabels()或ax.set_xticks()等方法修改坐标轴或字体的显示样式。
def ShowHeatMap_mic(matrix,labels):
colormap = plt.cm.RdBu
ylabels = labels
f, ax = plt.subplots(figsize=(15, 12))
ax.set_title('Matrix_mic')
sns.heatmap(matrix,
cmap=colormap,
ax=ax,
annot=True,
yticklabels=ylabels,
xticklabels=ylabels)
ax.set_xticklabels(labels, rotation=45,)
ax.set_xticks(range(0,20,1))
plt.show()
f.savefig('./相关系数矩阵_mic.png',dpi=300)
np.savetxt("./相关系数矩阵_mic.txt",matrix)
def ShowHeatMap_corr(dataframe,labels):
dataframe = dataframe.iloc[:,1:]
dataframe = dataframe.apply(lambda x:x.astype(float))
matrix=dataframe.corr()
colormap = plt.cm.RdBu
ylabels = labels
f, ax = plt.subplots(figsize=(15, 12))
ax.set_title('Matrix_corr')
sns.heatmap(matrix,
cmap=colormap,
ax=ax,
annot=True,
yticklabels=ylabels,
xticklabels=ylabels)
ax.set_xticklabels(labels, rotation=45,)
ax.set_xticks(range(0,20,1))
plt.show()
f.savefig('./相关系数矩阵_corr.png',dpi=300)
np.savetxt("./相关系数矩阵_corr.txt",matrix)