from rass_thetav import get_thetav from glob import glob from rv_utilities import add_colorbar, format_xaxis, format_yaxis homedir = os.path.expanduser('~') for r in range(8, 14): ''' case 14 does not have RASS obs ''' print 'case '+str(r) print '-----------------' fs = glob(homedir+'/RASS/c'+str(r).zfill(2)+'/*.cns') fs.sort() for n, f in enumerate(fs): T, Tc, Hgt, timestamp = mf.parse_rass(f) if n == 0: Tstack = T Hgtstack = Hgt ts = np.array(timestamp) else: Tstack = np.hstack((Tstack, T)) Hgtstack = np.hstack((Hgtstack, Hgt)) ts = np.hstack((ts, np.array(timestamp))) thetav = get_thetav(case=r, Tv_array=Tstack, hgt_array=Hgtstack, homedir=homedir) titletxt = 'Case{} ({})' fig, ax = plt.subplots(2, 1, sharex=True) im = ax[0].imshow(Tstack, interpolation='none', origin='lower') ax[0].set_title(titletxt.format(str(r), ts[0].strftime('%Y-%b')))