Python科学计算技巧积累四——双y轴图像绘制

双y轴图像具有单y轴图像没有的对比效果,在MATLAB中有plotyy函数可以实现,Python的实现方式没有MATLAB那样方便,不过实现效果却也不见得差。

以往我常用的绘图命令是
import matplotlib.pylab as plt # 导入绘图包
plt.figure() # 创建图像文件
plt.plot(...)  # 绘制图像
plt.show() # 显示图像

在双y轴制图中,绘图命令和以往不同,因而在此记录以备日后查阅。以一段程序为例
fig, ax1 = plt.subplots()  # 使用subplots()创建窗口

ax2 = ax1.twinx() # 创建第二个坐标轴

ax1.plot(pos_z, E_z, linewidth = 2)  # E_z是一组数据,不用在意

ax2.plot(pos_z, Enhance_z, linewidth = 3) # Ehance_z 是一组数据,不用在意

ax1.set_xlabel('position (nm)', fontsize = 16)  # fontsize使用方法和plt.xlabel()中一样

ax1.set_ylabel('|$E_{z}$| (V/m)', fontsize = 16)

ax2.set_ylabel('Enhancement', fontsize = 16)

ax1.set_xlim([0, max(pos_z)]) # 设置坐标轴范围的语句有所变化

ax1.set_ylim(0, max(E_z))

ax2.set_ylim([0, max(Enhance_z)])

plt.show()



图像最后的效果如下,由于E_z和Enhance_z是同一组数据,只不过其中一个被缩小了若干倍,所以最后两个图像重合了,只能看到一条线。

 

posted on 2018-03-09 17:42  莫水千流  阅读(569)  评论(0编辑  收藏  举报