在演示2维图时,如果需要保存有坐标和无坐标两种图片,就要在画坐标时做一些调整
Python一次绘图保存多张图和保存两张图的的思路是一致的,就是从无到有的构建,从要求最少的图开始画,然后保存,然后实现新的要求,这样往复。
通常来说,我们一般需要两张输出,一个是有坐标的,用于演示,一个是无坐标,无边框的纯图,可以用于图像处理或者后续的下游任务。下面代码以此为例说明。
import matplotlib as plt
import numpy as np
# 本示例为二维热力图,所以有三个维度的数据
def draw2d(x, y, z, name, path):
# 构建画布
fig = plt.figure(figsize=(32, 24), dpi=300)
X, Y = np.meshgrid(data[0], data[0]) # 因为是三维数据,需要构建网格
# 因为是热力图,所以要设置色条
c = plt.pcolormesh(X, Y, data[1], cmap="jet", shading='gouraud')
# 第一张图不需要坐标,所以对其置空
plt.axis("off")
plt.xticks([])
plt.yticks([])
save_path_ = path + "/" + name + "_2d_figure.png"
# bbox_inches 和 pad_inches是为了收紧边缘,让图像不留白边
fig.savefig(save_path_, bbox_inches='tight', pad_inches=0) # this