"""
Plot 2 contours and remove the last one
========================================
Use :func:`showvar()` to plot the wind as a filled contour plot and the temperature as grey contour lines on the same axes. 

"""

import pylab as pyl

import pygeode as pyg
import numpy as np
from pygeode.tutorial import t2

pyl.ioff()
# Load Temperature and zonal wind
u = t2.U(time='10 May 2002', pres=(1000,500)).mean(pyg.Lon)
T = t2.Temp(time='10 May 2002', pres=(1000,500)).mean(pyg.Lon)

# Change vertical axis to log(pressure) with a scale height of 7 km
u_H = u.replace_axes(pres=u.pres.logPAxis(H=7000))          
T_H = T.replace_axes(pres=T.pres.logPAxis(H=7000))     

ax = pyg.plot.AxesWrapper()

pyg.showvar(T_H, clines=10, colors='grey', axes=ax)  
ax_wrapper =pyg.showvar(u_H, min=10, cdelt=10, ndiv=3, fig=1, cmap=pyl.cm.BuPu_r, axes=ax) 

ax.setp(title = 'Plot of temperature and zonal mean wind')
pyl.ion()
ax_wrapper.render()

###########################################################
# ``ax_wrapper.axes[0].plots`` contains a list of all the plotted objects
#

print(*ax_wrapper.axes[0].plots, sep="\n")

###############################################################################################
# Remove the temperatue contours from the list and plot the zonal wind contours and colourbar only.
#

ax_wrapper.axes[0].plots = ax_wrapper.axes[0].plots[1:]

ax.setp(title = 'Plot of zonal mean wind')
ax_wrapper.render()

###############################################################################################

print(*ax_wrapper.axes[0].plots, sep="\n")