您的位置:首页 > 新闻 > 会展 > matplotlib绘制子图以及局部放大效果

matplotlib绘制子图以及局部放大效果

2024/10/11 3:11:16 来源:https://blog.csdn.net/Horizonhui/article/details/141484847  浏览:    关键词:matplotlib绘制子图以及局部放大效果

需求:绘制1*2的子图,子图1显示两个三角函数,子图2显示三个对数函数,子图2中对指定的区域进行放大。

绘图细节:

  • 每个子图中每个函数的数据存放到一个列表中,然后将每个子图的数据统一存到一个列表中,在循环绘制子图的过程中,可以循环取每个子图的数据以及子图数据中每一个函数的数据,以更好的应对数据量不同的情况。函数标签同理。
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
y3 = np.log2(x)  
y4 = np.log(x)
y5 = np.log10(x)sub = [[y1, y2], [y3, y4, y5]] #两个子图的数据
labels = [['sin', 'cos'], ['log2', 'loge', 'log10']] #数据标
  • 子图的位置以及需要放大的区间可以自行设置,设置的坐标范围以及子图大小不同,显示效果也不同,另外,此方法在进行子图数据绘制时,一定要将数据转为numpy格式,否则无法进行截取数据的逻辑运算。

完整代码

import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns
# 创建数据
x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)
y3 = np.log2(x)  
y4 = np.log(x)
y5 = np.log10(x)sub = [[y1, y2], [y3, y4, y5]] #两个子图的数据
labels = [['sin', 'cos'], ['log2', 'loge', 'log10']] #数据标签
function_name = ['三角函数', '对数函数']# 创建1*2的子图区域,每个子图包含两条线
fig, axes = plt.subplots(1, 2, figsize=(12, 6))
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号for i in range(2):  #遍历两个子图data = sub[i] #取出子图数据label = labels[i] #取出子图标签for j in range(len(data)):   #遍历每个子图数据axes[i].plot(x, data[j], label=label[j])axes[i].legend()axes[i].set_title(function_name[i])if i == 1: #为第二个子图设置放大ax_zoom = axes[i].inset_axes([0.5, 0.05, 0.3, 0.3])  # 放大图距离中点的位置,参数是[left, bottom, width, height]x_start, x_end = 0.5, 2  #需要放大的x轴区间y_start, y_end = -1, 1  #需要放大的y轴区间ax_zoom.set_xlim(x_start, x_end) #设置坐标轴范围ax_zoom.set_ylim(y_start, y_end)# 在局部放大图中绘制放大区域的三条线条# 此处一定要保证x和y是numpy数组for k, _ in enumerate(data):ax_zoom.plot(x[(x >= x_start) & (x <= x_end)], data[k][(x >= x_start) & (x <= x_end)])axes[i].indicate_inset_zoom(ax_zoom) # 使用indicate_inset函数在主图中标记放大区域,可不设置plt.tight_layout(w_pad=3) #紧凑布局, 子图左右间距为3
plt.show()

效果

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com