print 'Plotting part (a): spatiofreq' plt.subplot(yplots, xplots, 1) if res == 'native': lon4sf = lon lat4sf = lat elif res == 'make': lt1 = lat[0] lt2 = lat[-1] ln1 = lon[0] ln2 = lon[-1] lat4sf = np.arange(lt2, lt1 + extent, gsize) lat4sf = lat4sf[:: -1] # latitude has to be made the other way because of the negative numbers lon4sf = np.arange(ln1, ln2 + extent, gsize) m = blb.AfrBasemap2(lat4sf, lon4sf, drawstuff=True, prj='cyl', rsltn='l') allmask, img = stats.spatiofreq5(m, s, name, lat4sf, lon4sf, yrs, thesekeys, per=rate, clim=nos4cbar, \ savefig=False, flagonly=True, \ col='bw', frm_event=from_event,cbar='none',title='') m.drawcountries(color='k') m.drawcoastlines(color='k') if testingoutput: plt.savefig('tmpfig_a.png', dpi=150) print 'Plotting part (b): CB outlines' plt.subplot(yplots, xplots, 2) m = blb.AfrBasemap2(lat, lon, drawstuff=True, prj='cyl', rsltn='l') if plotshow == 'col5': nch = 5 cols = ['r', 'b', 'c', 'm', 'g'] if choose_cb == 'random':
# Count ndays in composite with same sign as mean mask_zeros=np.zeros((nlat,nlon),dtype=np.float32) for i in range(nlat): for j in range(nlon): count=len(np.where(anoms_signs[:,i,j]==comp_signs[i,j])[0]) perc=(float(count)/float(nsamp))*100 if perc>=perc_ag: mask_zeros[i,j]=1 else: mask_zeros[i,j]=0 # Plot print "Plotting for model "+name2 plt.subplot(yplots,xplots,cnt) if wplotdraw=='all': m = blb.AfrBasemap2(lat, lon, latsp,lonsp, drawstuff=True, prj='cyl', rsltn='l',\ fontdict={'fontsize':8,'fontweight':'normal'}) elif wplotdraw=='first': if cnt==1: m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \ fontdict={'fontsize': 8, 'fontweight': 'demibold'}) else: m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=False, prj='cyl', rsltn='l', \ fontdict={'fontsize': 8, 'fontweight': 'demibold'}) elif wplotdraw=='edges': x_remain= cnt % xplots if x_remain==1: m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \ fontdict={'fontsize': 8, 'fontweight': 'normal'}) else: m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \ fontdict={'fontsize': 8, 'fontweight': 'normal'},onlyedge='lon')
lat = np.arange(-70.0, 20.0, 10.0) lon = np.arange(-2.5, 120.0, 10.0) latsp = 20. # lat spacing lonsp = 25. # lon spacing ### Get directories bkdir = cwd + "/../../../../CTdata/" botdir = bkdir + "metbot_multi_dset/" figdir = botdir + "/histpaper_v2_figs/outlines_map/" my.mkdir_p(figdir) # Set up plot print "Setting up plot..." g, ax = plt.subplots(figsize=figdim) m = blb.AfrBasemap2(lat, lon, latsp, lonsp, drawstuff=True, prj='cyl', rsltn='l', \ fontdict={'fontsize': 12, 'fontweight': 'normal'}) # Redraw map m.drawcountries() m.drawcoastlines() # Plot MetBot full domain fulldom_lats = [0, 0, -60, -60] fulldom_lons = [7.5, 100.0, 100.0, 7.5] draw_screen_poly(fulldom_lats, fulldom_lons, m, ls='--', lw=3, edgecol='dodgerblue', zord=10)