图像显示子图

图像看不清，打算再左下角空白处画一个部分放大的子图

# -*- coding: utf-8 -*-
# 画放大子图
import h5py
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1.inset_locator import mark_inset
from mpl_toolkits.axes_grid1.inset_locator import inset_axes


# 准备数据
f = h5py.File("data.hdf5", "r")
keys = f.keys()
pulse = []
time = []
for e in f['pulse']:
    pulse.append(e)
for e in f['time']:
    time.append(e)

# x1 = time
# y1 = pulse
x = np.arange(len(pulse))
y = np.array(pulse)

# 绘图
fig, ax = plt.subplots(1, 1, figsize=(6, 10))
ax.plot(x, y, 'r', label='pulse') # 画线并添加图例legend
#  翻阅matplotlib的官方API手册发现ax根本没有title这个接口，正确的接口是set_title。所以，ax.title → ax.set_title
ax.set_title("Number of samples_pulse")

# 嵌入绘制局部放大图的坐标系
axins = inset_axes(ax, width="40%", height="30%", loc='lower left',
                   bbox_to_anchor=(0.5, 0.1, 1, 1),
                   bbox_transform=ax.transAxes)

# # 在子坐标系中绘制原始数据
axins.plot(x, y, 'r')

# 设置放大区间
zone_left = 2000
zone_right = 2800

# 坐标轴的扩展比例（根据实际数据调整）
x_ratio = 1.0  # x轴显示范围的扩展比例
y_ratio = 0.3  # y轴显示范围的扩展比例

# X轴的显示范围
xlim0 = x[zone_left] - (x[zone_right] - x[zone_left]) * x_ratio
xlim1 = x[zone_right] + (x[zone_right] - x[zone_left]) * x_ratio

# # Y轴的显示范围
y = np.hstack(y[zone_left:zone_right])
ylim0 = np.min(y) - (np.max(y) - np.min(y)) * y_ratio
ylim1 = np.max(y) + (np.max(y) - np.min(y)) * y_ratio

# 调整子坐标系的显示范围
axins.set_xlim(xlim0, xlim1)
axins.set_ylim(ylim0, ylim1)

# 建立父坐标系与子坐标系的连接线
# loc1 loc2: 坐标系的四个角
# 1 (右上) 2 (左上) 3(左下) 4(右下)
mark_inset(ax, axins, loc1=3, loc2=1, fc="none", ec='k', lw=1)

# 显示
plt.show()

参考链接