python科研绘图：左上角RMSE，右下角R2,并画出双色带

CNU-ZQQ

已于 2022-05-17 08:19:06 修改

阅读量3k

点赞数 1

分类专栏：科研绘图文章标签： python

于 2021-09-05 14:04:50 首次发布

本文链接：https://blog.csdn.net/qq_34907666/article/details/120113914

版权

科研绘图专栏收录该内容

4 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用Python进行科研绘图，特别是如何创建一个包含右上角RMSE和左下角R2指标的双色矩阵图。示例代码展示了如何生成随机数据并用matplotlib库绘制图像，同时添加颜色条和坐标标签，最后保存为PDF文件。

摘要由CSDN通过智能技术生成

1.结果图

在这里插入图片描述

2.直接上代码

import numpy as np
import matplotlib.pyplot as plt
import random
#自己做的数据，可自行调整
label1=[‘a’,‘b’,‘c’,‘d’,‘e’,‘f’,‘g’,‘h’]
arr1=np.zeros((len(label1),len(label1)))*np.nan
arr2=np.zeros((len(label1),len(label1)))*np.nan
for i in range(1,len(label1)+1):
for j in range(i):
if i-1==j:continue
arr1[i-1,j]=random.random()
arr2[j,i-1]=random.random()*10

#绘图部分
fig = plt.figure(figsize=(5,5),dpi=600)# #(宽，高)
ax = fig.add_subplot(111)
ax.xaxis.tick_top()
im1=plt.imshow(arr1,vmin=0, vmax=1, cmap=‘summer_r’)
im2=plt.imshow(arr2,vmin=0, vmax=10,cmap=‘autumn_r’)
plt.colorbar(im2,shrink=0.7,spacing=‘uniform’,pad=0.01,aspect=12,ax=ax)#
plt.colorbar(im1,shrink=0.7,spacing=‘uniform’,pad=0.01,aspect=12,ax=ax,orientation=‘horizontal’)#
plt.xticks(range(8), labels=label1)#,rotation=330（如果文字比较长可以旋转一下）
plt.yticks(range(8), labels=label1)
plt.savefig(‘temp.pdf’)
plt.show()
plt.close()