Merge pull request #1642 from knutfrode/dev

[run-ex] Updated example to illustrate vertical velocity correction

Author: knutfrode

examples/example_vertical_velocity_correction.py

@@ -54,16 +54,18 @@
 o3.run(duration=timedelta(hours=24), time_step=1800)
 
 
-plt.subplot(3,1,1)
-o.result.z.plot.line(x='time', add_legend=False, color='k')
-o2.result.z.plot.line(x='time', add_legend=False, color='r')
-o3.result.z.plot.line(x='time', add_legend=False, color='g')
+fig = plt.figure(figsize=(10, 10))
+gs = fig.add_gridspec(5, 1)
+ax1 = fig.add_subplot(gs[0:3, 0])
+o.result.z.plot.line(ax=ax1, x='time', add_legend=False, color='k')
+o2.result.z.plot.line(ax=ax1, x='time', add_legend=False, color='r')
+o3.result.z.plot.line(ax=ax1, x='time', add_legend=False, color='g')
 plt.plot([], [], color='k', label='No stretching or correction')
 plt.plot([], [], color='r', label='Column stretching')
 plt.plot([], [], color='g', label='Vertical velocity correction')
 plt.legend()
-plt.subplot(3,1,2)
-o.result.sea_floor_depth_below_sea_level.plot.line(x='time', add_legend=False, color='k')
-plt.subplot(3,1,3)
-o.result.upward_sea_water_velocity.plot.line(x='time', add_legend=False, color='k')
+ax2 = fig.add_subplot(gs[3, 0])
+o.result.sea_floor_depth_below_sea_level.plot.line(ax=ax2, x='time', add_legend=False, color='k')
+ax3 = fig.add_subplot(gs[4, 0])
+o.result.upward_sea_water_velocity.plot.line(ax=ax3, x='time', add_legend=False, color='k')
 plt.show()