Search code examples
matplotlibmatplotlib-basemap

How to draw rectangles on a Basemap


I'm looking for a way to plot filled rectangles on a Basemap. I could easily draw the rectangle's edges using the drawgreatcircle method, but I cannot find a way to actually fill these rectangles (specifying color and alpha).


Solution

  • You can add a matplotlib.patches.Polygon() directly to your axes. The question is whether you want your rectangles defined the plot coordinates (straight lines on the plot) or in map coordinates (great circles on the plot). Either way, you specify vertices in map coordinates and then transform them to plot coordinates by calling the Basemap instance (m() in the below example), build a Polygon yourself, and add it manually to the axes to be rendered.

    For rectangles defined in plot coordinates, here's an example:

    from mpl_toolkits.basemap import Basemap
    import numpy as np
    import matplotlib.pyplot as plt
    from matplotlib.patches import Polygon
    
    def draw_screen_poly( lats, lons, m):
        x, y = m( lons, lats )
        xy = zip(x,y)
        poly = Polygon( xy, facecolor='red', alpha=0.4 )
        plt.gca().add_patch(poly)
    
    lats = [ -30, 30, 30, -30 ]
    lons = [ -50, -50, 50, 50 ]
    
    m = Basemap(projection='sinu',lon_0=0)
    m.drawcoastlines()
    m.drawmapboundary()
    draw_screen_poly( lats, lons, m )
    
    plt.show()
    

    For rectangles defined in map coordinates, use the same approach, but interpolate your line in map space before transforming to plot coordinates. For each line segment, you'll have to do:

    lats = np.linspace( lat0, lat1, resolution )
    lons = np.linspace( lon0, lon1, resolution )
    

    Then transform these map coordinates to plot coordinates (as above, with m()) and again create a Polygon with the plot coordinates.