【python绘图】绘制局部放大图

绘制效果如下：

左图为原图，右图为局部放大图。选中原图中的一小部分矩形区域框出，然后放大，将原图和放大图都绘制在同一个画布上，并将对角线采用直线连接。

代码可以直接使用，只需要替换一些参数即可，部分参数设置解释：

color = (255, 0, 0)                               # 颜色
folder_path = r"originals_path"           # 待处理图像文件目录 
folder_path2 = r"rectangles_path"      # 处理后图像保存目录
resize = (700,457)                              # 请根据原图像素大小自行设置
x1, y1 = 110, 140                                # 原图中局部放大区域左上角坐标
x2, y2 = 210, 210                                # 原图中局部放大区域右下角坐标
linewidth = 2                                       # 绘制线粗细
zoom_array = [0.55, 0.10, 0.4, 0.4]    # 即[A,B,C,D]

如下图所示：A、B用于固定局部放大图左下角的位置，C、D指定局部放大图大小。这里的A、B、C、D都是相对于整个画布的比例。A越大，局部放大图越靠近右边；B越大，越靠近上方；C、D越大，局部放大图相对于整个画布越大。请根据实际情况多多尝试，以找出合适的取值。

绘制两条连接线的参数与代码：

# 绘制线连接原始图像和放大图
line = ConnectionPatch(xyA=(x1, y2), xyB=(0, 0), coordsA="data", coordsB="axes fraction", axesA=ax, axesB=zoom_ax, color=color, linewidth=linewidth)
fig.add_artist(line)

# xyA：原图中局部放大区域的顶点坐标

# xyB：局部放大图的顶点坐标

# 这里的意思是连接原图中的(x1, y2)点与局部放大图的(0, 0)点

根据局部放大图的位置，这部分参数可能需要调整。具体取值和连接效果可参考下图：

完整代码如下，可以根据需求省略或补充部分代码：

import matplotlib.pyplot as plt
import matplotlib.patches as patches
from matplotlib.patches import ConnectionPatch
import numpy as np
from PIL import Image
import os

# 待修改参数
color = (255, 0, 0)
folder_path = r"originals_path"
folder_path2 = r"rectangles_path"
resize = (700,457)
# 定义局部放大区域的坐标
x1, y1 = 110, 140
x2, y2 = 210, 210
linewidth = 2
zoom_array = [0.55, 0.10, 0.4, 0.4]   # x y 子图左下角相对于整个画幅的位置， 后两位子图大小


def normalize_color(color):
    normalized_color = tuple(channel / 255.0 for channel in color)
    return normalized_color
color = normalize_color(color)

# 遍历文件夹中的每个图像文件
for filename in os.listdir(folder_path):
    if filename.endswith(".png"):
        image_path = os.path.join(folder_path, filename)
        image = Image.open(image_path)
        image = image.resize(resize)
        image_array = np.array(image)


        # 获取局部放大区域的图像数据
        zoomed_region = image_array[y1:y2, x1:x2]
        fig, ax = plt.subplots()

        # 绘制原始图像
        ax.imshow(image_array)
        ax.axis('off')

        # 绘制局部放大区域的边界框
        rect = patches.Rectangle((x1, y1), x2-x1, y2-y1, linewidth=linewidth, edgecolor=color, facecolor='none')
        ax.add_patch(rect)

        # 创建放大图
        zoom_ax = fig.add_axes(zoom_array)
        zoom_ax.imshow(zoomed_region)
        # 设置放大图的边框，并增加线宽
        zoom_ax.spines['top'].set_color(color)
        zoom_ax.spines['top'].set_linewidth(linewidth)
        zoom_ax.spines['bottom'].set_color(color)
        zoom_ax.spines['bottom'].set_linewidth(linewidth)
        zoom_ax.spines['left'].set_color(color)
        zoom_ax.spines['left'].set_linewidth(linewidth)
        zoom_ax.spines['right'].set_color(color)
        zoom_ax.spines['right'].set_linewidth(linewidth)
        # 隐藏放大图的坐标轴
        zoom_ax.set_xticks([])
        zoom_ax.set_yticks([])


        # 绘制线连接原始图像和放大图
        line = ConnectionPatch(xyA=(x1, y2), xyB=(0, 0), coordsA="data", coordsB="axes fraction", axesA=ax, axesB=zoom_ax, color=color, linewidth=linewidth-1)
        fig.add_artist(line)
        line2 = ConnectionPatch(xyA=(x2, y1), xyB=(1, 1), coordsA="data", coordsB="axes fraction", axesA=ax, axesB=zoom_ax, color=color, linewidth=linewidth-1)
        fig.add_artist(line2)

        # plt.show()
        # 保存图像
        save_path = os.path.join(folder_path2, f"zoomed_{filename}")
        plt.savefig(save_path,transparent=True,dpi=300,bbox_inches='tight')