Python画2维图同时保存无、有坐标图等多张图的方式_python怎样保存不带坐标的图像-优快云博客

本文链接：https://blog.youkuaiyun.com/kecilimu/article/details/140433617

在演示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 image is used for analysis, the follow image is used for show
		# 第二张需要坐标，特别注意，如果前面把，x,y的坐标都设置成了“[]”，那么这里给出具体的坐标向量
    plt.axis("on")
    plt.title(name, {'fontsize': 64, 'fontweight': 'bold'})
    plt.xlabel("X(cm)", {'fontsize': 44})
    plt.ylabel("Y(cm)", {'fontsize': 44})
    xt = np.around(np.arange(-3.5, 3.5, 0.5), decimals=1)
    plt.xticks(xt)
    plt.yticks(xt)
    plt.xticks(fontsize=16)
    plt.yticks(fontsize=16)
    cb = plt.colorbar(c)  # 画出色条
    cb.ax.tick_params(labelsize=24)
    save_path = path + "/" + name + "_2d.png"
    plt.savefig(save_path)
    # plt.show()
    plt.close()
    return save_path_