use Basemap colorbar method.

Author: Jeff Whitaker

examples/barb_demo.py

@@ -41,9 +41,7 @@
 # plot barbs.
 m.barbs(xv,yv,udat,vdat,length=6,barbcolor='k',flagcolor='r',linewidth=0.5)
 # plot colorbar for pressure
-cax = plt.axes([0.875, 0.1, 0.05, 0.8]) # setup colorbar axes.
-plt.colorbar(cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(pad='12%') # draw colorbar
 # draw coastlines
 m.drawcoastlines()
 # draw parallels
@@ -72,9 +70,7 @@
 # plot barbs.
 m.barbs(xv,yv,udat,vdat,length=6,barbcolor='k',flagcolor='r',linewidth=0.5)
 # plot colorbar for pressure
-cax = plt.axes([0.875, 0.1, 0.05, 0.8]) # setup colorbar axes.
-plt.colorbar(cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(pad='12%') # draw colorbar
 # draw coastlines
 m.drawcoastlines()
 # draw parallels

examples/contour_demo.py

@@ -23,12 +23,7 @@
 CS1 = m.contour(x,y,hgt,15,linewidths=0.5,colors='k')
 # fill between contour lines.
 CS2 = m.contourf(x,y,hgt,CS1.levels,cmap=plt.cm.jet,extend='both')
-# setup colorbar axes instance.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.075, b, 0.05, h]) # setup colorbar axes
-plt.colorbar(CS2, drawedges=True, cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(CS2) # draw colorbar
 # draw coastlines and political boundaries.
 m.drawcoastlines()
 m.drawmapboundary()
@@ -50,11 +45,7 @@
 x, y = m(lons, lats)
 CS1 = m.contour(x,y,hgt,15,linewidths=0.5,colors='k')
 CS2 = m.contourf(x,y,hgt,CS1.levels,cmap=plt.cm.jet,extend='both')
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.075, b, 0.05, h]) # setup colorbar axes
-plt.colorbar(CS2, drawedges=True, cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(CS2) # draw colorbar
 # draw coastlines and political boundaries.
 m.drawcoastlines()
 m.drawmapboundary()
@@ -76,11 +67,7 @@
 x, y = m(lons, lats)
 CS1 = m.contour(x,y,hgt,15,linewidths=0.5,colors='k')
 CS2 = m.contourf(x,y,hgt,CS1.levels,cmap=plt.cm.jet,extend='both')
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.075, b, 0.05, h]) # setup colorbar axes
-plt.colorbar(CS2, drawedges=True, cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(CS2) # draw colorbar
 # draw coastlines and political boundaries.
 m.drawcoastlines()
 m.drawmapboundary()
@@ -102,11 +89,7 @@
 x, y = m(lons, lats)
 CS1 = m.contour(x,y,hgt,15,linewidths=0.5,colors='k')
 CS2 = m.contourf(x,y,hgt,CS2.levels,cmap=plt.cm.jet,extend='both')
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.075, b, 0.05, h]) # setup colorbar axes
-plt.colorbar(CS2, drawedges=True, cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(CS2,pad='12%') # draw colorbar
 # draw coastlines and political boundaries.
 m.drawcoastlines()
 m.drawmapboundary()
@@ -129,11 +112,7 @@
 x, y = m(lons, lats)
 CS1 = m.contour(x,y,hgt,15,linewidths=0.5,colors='k')
 CS2 = m.contourf(x,y,hgt,CS1.levels,cmap=plt.cm.jet,extend='both')
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.075, b, 0.05, h]) # setup colorbar axes
-plt.colorbar(CS2, drawedges=True, cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(CS2) # draw colorbar
 # draw coastlines and political boundaries.
 m.drawcoastlines()
 m.fillcontinents()

examples/geos_demo_2.py

@@ -46,8 +46,6 @@
 # be outside map projection region.
 m.drawmeridians(np.arange(10,76,5), labels=[0,0,1,0], color=overlay_color)
 m.drawparallels(np.arange(-90,90,5), labels=[1,0,0,0], color=overlay_color)
-# add a colobar
-#plt.colorbar()
 # add timestamp and save
 fig = plt.gcf()
 fig.text(x=0.275, y=0.025, s=u'Meteosat-9 VIS 0.6 channel - 12:00 UTC 04/06/2007\n    \N{COPYRIGHT SIGN} EUMETSAT 2007',

examples/panelplot.py

@@ -28,11 +28,7 @@
 CS = mnh.contour(xnh,ynh,hgt,15,linewidths=0.5,colors='k')
 CS = mnh.contourf(xnh,ynh,hgt,15,cmap=plt.cm.Spectral)
 # colorbar on bottom.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l, b-0.05, w, 0.025]) # setup colorbar axes
-plt.colorbar(cax=cax, orientation='horizontal',ticks=CS.levels[0::4]) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+mnh.colorbar(location='bottom',pad='12%',ticks=CS.levels[0::4])
 mnh.drawcoastlines(linewidth=0.5)
 delat = 30.
 circles = np.arange(0.,90.,delat).tolist()+\
@@ -51,11 +47,7 @@
 CS = msh.contour(xsh,ysh,hgt,15,linewidths=0.5,colors='k')
 CS = msh.contourf(xsh,ysh,hgt,15,cmap=plt.cm.Spectral)
 # colorbar on bottom.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l, b-0.05, w, 0.025]) # setup colorbar axes
-plt.colorbar(cax=cax,orientation='horizontal',ticks=MultipleLocator(320)) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+msh.colorbar(location='bottom',pad='12%',ticks=MultipleLocator(320))
 msh.drawcoastlines(linewidth=0.5)
 msh.drawparallels(circles,labels=[1,0,0,0])
 msh.drawmeridians(meridians,labels=[1,0,0,1])
@@ -74,11 +66,7 @@
 CS = mnh.contour(xnh,ynh,hgt,15,linewidths=0.5,colors='k')
 CS = mnh.contourf(xnh,ynh,hgt,15,cmap=plt.cm.RdBu)
 # colorbar on right
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.025, b, 0.025, h]) # setup colorbar axes
-plt.colorbar(cax=cax, ticks=MultipleLocator(160), format='%4i') # draw colorbar
-plt.axes(ax)  # make the original axes current again
+mnh.colorbar(location='right',ticks=MultipleLocator(160))
 mnh.drawcoastlines(linewidth=0.5)
 mnh.drawparallels(circles,labels=[1,0,0,0])
 mnh.drawmeridians(meridians,labels=[1,0,0,1])
@@ -89,11 +77,7 @@
 CS = msh.contour(xsh,ysh,hgt,15,linewidths=0.5,colors='k')
 CS = msh.contourf(xsh,ysh,hgt,15,cmap=plt.cm.RdBu)
 # colorbar on right.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.025, b, 0.025, h]) # setup colorbar axes
-plt.colorbar(cax=cax, ticks=MultipleLocator(160), format='%4i') # draw colorbar
-plt.axes(ax)  # make the original axes current again
+msh.colorbar(location='right',ticks=MultipleLocator(160))
 msh.drawcoastlines(linewidth=0.5)
 msh.drawparallels(circles,labels=[1,0,0,0])
 msh.drawmeridians(meridians,labels=[1,0,0,1])

examples/plotmap_masked.py

@@ -30,8 +30,8 @@
 topodat,x,y = m.transform_scalar(topoin,lonsin,latsin,nx,ny,returnxy=True)
 # create the figure.
 fig=plt.figure(figsize=(8,8))
-# add an axes, leaving room for colorbar on the right.
-ax = fig.add_axes([0.1,0.1,0.7,0.7])
+# add an axes.
+ax = fig.add_axes([0.1,0.1,0.8,0.8])
 # associate this axes with the Basemap instance.
 m.ax = ax
 # make topodat a masked array, masking values lower than sea level.
@@ -41,11 +41,7 @@
 palette.set_bad('aqua', 1.0)
 # plot image over map with imshow.
 im = m.imshow(topodatm,palette,norm=colors.normalize(vmin=0.0,vmax=3000.0,clip=False))
-# setup colorbar axes instance.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.075, b, 0.05, h])
-plt.colorbar(im,cax=cax) # draw colorbar
+m.colorbar(im,pad='12%') # draw colorbar
 # plot blue dot on boulder, colorado and label it as such.
 xpt,ypt = m(-104.237,40.125) 
 m.plot([xpt],[ypt],'bo') 

examples/plotprecip.py

@@ -32,7 +32,7 @@
             llcrnrlon=loncorners[0],urcrnrlon=loncorners[2],\
             rsphere=6371200.,resolution='l',area_thresh=10000)
 # create figure
-fig = plt.figure(figsize=(6,8.5))
+fig = plt.figure(figsize=(8.5,11))
 plt.subplot(211)
 ax = plt.gca()
 # draw coastlines, state and country boundaries, edge of map.
@@ -53,13 +53,9 @@
 # draw filled contours.
 clevs = [0,1,2.5,5,7.5,10,15,20,30,40,50,70,100,150,200,250,300,400,500,600,750]
 cs = m.contourf(x,y,data,clevs,cmap=cm.s3pcpn)
-# new axis for colorbar.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.025, b, 0.025, h]) # setup colorbar axes
 # draw colorbar.
-cb = plt.colorbar(cs, cax, format='%g', ticks=clevs, drawedges=False) 
-plt.axes(ax)  # make the original axes current again
+cbar = m.colorbar(cs,location='bottom',pad="10%")
+cbar.set_label('mm')
 # plot title
 plt.title(plottitle+'- contourf',fontsize=10)
 
@@ -79,16 +75,12 @@
 # colormap to linear version of the precip colormap.
 im2 = copy.copy(im)
 im2.set_cmap(cm.s3pcpn_l)
-# new axis for colorbar.
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.025, b, 0.025, h]) # setup colorbar axes
-# using im2, not im (hack to prevent colors from being
+# draw colorbar using im2, not im (hack to prevent colors from being
 # too compressed at the low end on the colorbar - results
 # from highly nonuniform colormap)
-cb = plt.colorbar(im2, cax, format='%d') # draw colorbar
-plt.axes(ax)  # make the original axes current again
-# reset colorbar tick labels (hack to get
+cb = m.colorbar(im2,location='bottom',pad="10%")
+cb.set_label('mm')
+# reset colorbar tick labels.
 cb.set_ticks(np.linspace(clevs[0],clevs[-1],len(clevs)))
 cb.set_ticklabels(['%g' % clev for clev in clevs])
 # plot title

examples/plotsst.py

@@ -1,24 +1,16 @@
-from mpl_toolkits.basemap import Basemap, date2index, num2date
-from netCDF4 import Dataset as NetCDFFile
+from mpl_toolkits.basemap import Basemap
+from netCDF4 import Dataset, date2index, num2date
 import numpy as np
 import matplotlib.pyplot as plt
-import sys, datetime
-# read in sea-surface temperature and ice data
-# can be a local file, a URL for a remote opendap dataset,
-if len(sys.argv) == 1:
-    date = '20071215'
-else:
-    date = sys.argv[1]
-# convert datestring to datetime object.
-date = datetime.datetime(int(date[0:4]),int(date[4:6]),int(date[6:8]))
-print date
+import datetime
+# create datetime object for desired time
+date = datetime.datetime(2007,12,15,0)
 # open dataset.
 dataset =\
-NetCDFFile('http://nomads.ncdc.noaa.gov/thredds/dodsC/oisst2/totalAmsrAgg')
+Dataset('http://nomads.ncdc.noaa.gov/thredds/dodsC/oisst2/totalAmsrAgg')
 # find index of desired time.
 time = dataset.variables['time']
 nt = date2index(date, time, calendar='standard')
-print num2date(time[nt],time.units, calendar='standard')
 # read sst.  Will automatically create a masked array using
 # missing_value variable attribute.
 sst = dataset.variables['sst'][nt]
@@ -29,18 +21,20 @@
 lons = dataset.variables['lon'][:]
 # shift lats, lons so values represent edges of grid boxes
 # (as pcolor expects).
-delon = lons[1]-lons[0]
-delat = lats[1]-lats[0]
+delon = lons[1]-lons[0]; delat = lats[1]-lats[0]
 lons = (lons - 0.5*delon).tolist()
 lons.append(lons[-1]+delon)
 lons = np.array(lons,np.float64)
 lats = (lats - 0.5*delat).tolist()
 lats.append(lats[-1]+delat)
 lats = np.array(lats,np.float64)
-# create Basemap instance for mollweide projection.
+# creat figure, axes instances.
+fig = plt.figure()
+ax = fig.add_axes([0.05,0.05,0.9,0.9])
+# create Basemap instance for Robinson projection.
 # coastlines not used, so resolution set to None to skip
 # continent processing (this speeds things up a bit)
-m = Basemap(projection='moll',lon_0=lons.mean(),lat_0=0,resolution=None)
+m = Basemap(projection='robin',lon_0=lons.mean(),resolution=None)
 # compute map projection coordinates of grid.
 x, y = m(*np.meshgrid(lons, lats))
 # draw line around map projection limb.
@@ -53,8 +47,8 @@
 # draw parallels and meridians, but don't bother labelling them.
 m.drawparallels(np.arange(-90.,120.,30.))
 m.drawmeridians(np.arange(0.,420.,60.))
-# draw horizontal colorbar.
-plt.colorbar(im1,orientation='horizontal')
-# display the plot with a title.
+# add colorbar
+cb = m.colorbar(im1,"bottom", size="5%", pad="2%")
+# add a title.
 plt.title('SST and ICE analysis for %s'%date)
 plt.show()

examples/quiver_demo.py

@@ -36,9 +36,7 @@
 # plot wind vectors over map.
 Q = m.quiver(x,y,urot,vrot) #or specify, e.g., width=0.003, scale=400)
 qk = plt.quiverkey(Q, 0.95, 1.05, 25, '25 m/s', labelpos='W')
-cax = plt.axes([0.875, 0.1, 0.05, 0.7]) # setup colorbar axes.
-plt.colorbar(cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(pad='12%') # draw colorbar
 m.drawcoastlines()
 m.drawcountries()
 # draw parallels
@@ -72,9 +70,7 @@
 # plot wind vectors over map.
 Q = m.quiver(xv,yv,udat,vdat) #or specify, e.g., width=0.003, scale=400)
 qk = plt.quiverkey(Q, 0.95, 1.05, 25, '25 m/s', labelpos='W')
-cax = plt.axes([0.875, 0.1, 0.05, 0.7]) # setup colorbar axes.
-plt.colorbar(cax=cax) # draw colorbar
-plt.axes(ax)  # make the original axes current again
+m.colorbar(pad='12%') # draw colorbar
 m.drawcoastlines()
 m.drawcountries()
 # draw parallels

examples/randompoints.py

@@ -30,7 +30,7 @@
 # (otherwise they are covered up when continents are filled)
 m.scatter(x,y,25,z,cmap=plt.cm.jet,marker='o',edgecolors='none',zorder=10) 
 # plot colorbar for markers.
-plt.colorbar()
+m.colorbar()
 # create a list of strings containing z values
 # or, plot actual numbers as color-coded text strings.
 #zn = [ '%2i' % zz for zz in z ]

examples/testgdal.py

@@ -33,8 +33,8 @@
 # plot DEM image on map.
 im = m.imshow(topodat,cmap=cm.GMT_haxby_r)
 # draw meridians and parallels.
-m.drawparallels(np.arange(20,71,10),labels=[1,0,1,0])
-m.drawmeridians(np.arange(-120,-40,10),labels=[0,1,0,1])
+m.drawparallels(np.arange(20,71,10),labels=[1,0,0,0])
+m.drawmeridians(np.arange(-120,-40,10),labels=[0,0,0,1])
 # plot state boundaries from shapefile using ogr.
 g = ogr.Open ("st99_d00.shp")
 L = g.GetLayer(0) # data is in 1st layer.
@@ -58,13 +58,7 @@
                 lats = [g.GetY(i) for i in range(g.GetPointCount())]
                 x, y = m(lons,lats)
                 m.plot(x,y,'k')
-# new axis for colorbar.
-ax = plt.gca()
-pos = ax.get_position()
-l, b, w, h = pos.bounds
-cax = plt.axes([l+w+0.05, b, 0.025, h]) # setup colorbar axes
 # draw colorbar.
-plt.colorbar(im, cax=cax)
-plt.axes(ax)  # make the original axes current again
+m.colorbar(im)
 plt.title(gd.GetDescription()+' with state boundaries from '+g.GetName(),y=1.05)
 plt.show()