In [ ]:
%pylab notebook
import glob
import matplotlib.pyplot as plt
from charistools.hypsometry import Hypsometry

In [ ]:
list = glob.glob("/Users/brodzik/projects/CHARIS/derived_hypsometries/evapotranspiration/mod16/monthly/AM_*fullbasinmasks*txt")
list

In [ ]:
fig, ax = plt.subplots(5, 2, figsize=(14,16))

In [ ]:
basins = ['AM', 'BR', 'GA_v01', 'IN_v01', 'SY_v01']
vmax = [0.75, 5.5, 19.0, 5.5, 0.8]
for i, basin in enumerate(basins):
    list = sort(glob.glob("/Users/brodzik/projects/CHARIS/derived_hypsometries/evapotranspiration/mod16/monthly/" + basin + "_*fullbasinmasks*txt"))
    print(list)
    snowhyps = Hypsometry(filename=list[1])
    snowhyps.data = snowhyps.data['2001-01-01':'2001-12-31']
    hyps = Hypsometry(filename=list[0])
    hyps.data = hyps.data['2001-01-01':'2001-12-31']
    snowhyps.imshow(ax=ax[i,0], title=basin + ' (snowy) ET', cmap='Greens_r', 
                    xlabel='Date', dateFormat='%b', ylabel='Elevation ($m$)',
                    vmax=vmax[i])
    hyps.imshow(ax=ax[i,1], title=basin + ' ET', cmap='Greens_r', 
                xlabel='Date', dateFormat='%b', ylabel='Elevation ($m$)',
                vmax=vmax[i])
fig.tight_layout()

In [ ]:
#ax[0,0].plot(hyps.data['2001-01-01':'2001-01-01'])
snowhyps.imshow(ax=ax[0,0], title='AM (snowy)', cmap='Greys_r', 
                xlabel='Date', dateFormat='%b', 
                ylabel='Elevation ($m$)')
hyps.imshow(ax=ax[0,1], title='AM', cmap='Greys_r', 
            xlabel='Date', dateFormat='%b', 
            ylabel='Elevation ($m$)')

In [ ]:
hyps.data

In [ ]:
fig.savefig("/Users/brodzik/ipython_notebooks/charis/ET.2001.png")

In [ ]:
fig, ax = plt.subplots(1)
list = glob.glob("/Users/brodzik/projects/CHARIS/derived_hypsometries/evapotranspiration/mod16/monthly/AM_*fullbasinmasks*txt")
#    snowhyps = Hypsometry(filename=list[0])
#    snowhyps.data = snowhyps.data['2001-01-01':'2001-12-31']
#    hyps = Hypsometry(filename=list[1])
#    hyps.data = hyps.data['2001-01-01':'2001-12-31']
#    snowhyps.imshow(ax=ax[i,0], title=basin + ' (snowy) ET', cmap='Greens_r', 
#                    xlabel='Date', dateFormat='%b', ylabel='Elevation ($m$)')
#    hyps.imshow(ax=ax[i,1], title=basin + ' ET', cmap='Greens_r', 
#                xlabel='Date', dateFormat='%b', ylabel='Elevation ($m$)')
fig.tight_layout()