如何自动优化zoomed_inset_axes的zoom参数，最大化插入图且无重叠？

问题描述 #

我正在创建带放大插入图的绘图，数据类型多样，没法在程序启动前预知数据形态。希望放大插入图尽可能大，还不能和绘图其他元素重叠。示例代码里用了zoom=2，但这个值太小，手动调太麻烦，有没有办法自动找到能最大化插入图尺寸且不重叠的zoom参数？

解决方案 #

核心思路是根据主图的可用空间和插入图的数据范围，反向计算出最大可行的zoom值，避免手动调试。下面是修改后的代码，包含自动计算逻辑：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1.inset_locator import zoomed_inset_axes
from mpl_toolkits.axes_grid1.inset_locator import mark_inset

fig, ax = plt.subplots()

# 生成随机模拟数据（模拟未知数据形态）
xin = np.linspace(0, np.random.uniform(.5, 4), 1000)    
x_samples = np.random.uniform(0.9, 1.1, (1, 1000)) * np.sqrt(xin[:, np.newaxis])

ax.fill_between(xin, x_samples.min(1), x_samples.max(1))

# 定义插入图要放大的数据范围
inset_x_min, inset_x_max = 0.05, 0.1
idx = np.logical_and(xin > inset_x_min, xin < inset_x_max)
inset_y_min = x_samples.min(1)[idx].min()
inset_y_max = x_samples.max(1)[idx].max()

# 获取主图的坐标范围和像素尺寸
ax_xlim = ax.get_xlim()
ax_ylim = ax.get_ylim()
ax_pix_extent = ax.get_window_extent()
ax_width_pix = ax_pix_extent.width
ax_height_pix = ax_pix_extent.height

# 计算插入图数据范围在主图中的相对占比
x_range_ratio = (inset_x_max - inset_x_min) / (ax_xlim[1] - ax_xlim[0])
y_range_ratio = (inset_y_max - inset_y_min) / (ax_ylim[1] - ax_ylim[0])

# 计算最大可行zoom值：
# 1. 假设插入图放置区域占主图的40%空间（可根据loc参数调整）
# 2. 预留10%边距避免和主图元素/标记线重叠
max_zoom_x = (0.4 * ax_width_pix) / (x_range_ratio * ax_width_pix) * 0.9
max_zoom_y = (0.4 * ax_height_pix) / (y_range_ratio * ax_height_pix) * 0.9
# 取x、y方向的最小值，保证插入图完全适配可用空间
optimal_zoom = min(max_zoom_x, max_zoom_y)

# 创建插入图并配置参数
axins = zoomed_inset_axes(ax, zoom=optimal_zoom, loc='upper left')
axins.fill_between(xin, x_samples.min(1), x_samples.max(1))
axins.set_xlim(inset_x_min, inset_x_max)
axins.set_ylim(inset_y_min, inset_y_max)
axins.set_xticks([])
axins.set_yticks([])

mark_inset(ax, axins, loc1=4, loc2=3, fc="none", ec="0.5")

plt.savefig('hey.png')
plt.clf()

逻辑说明 #

• 先通过get_xlim()、get_ylim()获取主图的坐标范围，get_window_extent()获取主图的像素尺寸
• 计算插入图数据范围在主图中的比例，再结合插入图放置区域的可用空间（这里针对upper left位置取40%空间，若换其他位置比如lower right，直接调整这个比例即可）
• 预留10%的边距避免插入图和主图元素、标记线重叠，取x、y方向的最小zoom值，确保插入图在两个方向都能完全放入可用空间

内容的提问来源于stack exchange，提问作者kilojoules