2025年3月24日 星期一 甲辰(龙)年 月廿三 设为首页 加入收藏
rss
您当前的位置:首页 > 计算机 > 编程开发 > Python

图像显示子图

时间:04-17来源:作者:点击数:37

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

在这里插入图片描述
  • # -*- 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()
在这里插入图片描述

参考链接

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门