日志

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

已有 13359 次阅读| 2015-5-9 11:21 |个人分类:编程语言

双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是同一组数据，只不过其中一个被缩小了若干倍，所以最后两个图像重合了，只能看到一条线。