map_locations.py 文件源码

def main():
    args  = get_arguments()

    path = load_path(args.eclipse_path_data)

    # The funny constants define the lower left and upper right
    # corners (lat and lon) of the map boundary
    map = Basemap(llcrnrlat=22, llcrnrlon=-119, urcrnrlat=49, urcrnrlon=-64,
                  projection='lcc',lat_1=33,lat_2=45,lon_0=-95,
                  resolution='l')
    base_color = 'white'
    border_color = 'lightgray'
    boundary_color = 'gray'
    map.fillcontinents(color=base_color, lake_color=border_color)
    map.drawstates(color=border_color)
    map.drawcoastlines(color=border_color)
    map.drawcountries(color=border_color)
    map.drawmapboundary(color=boundary_color)

    points = pickle.load(open(args.inside_filename))
    lats = []
    lons = []
    colors = []
    for point in points:
        lat = point[0]
        lon = point[1]
        colors.append('r')
        lats.append(lat)
        lons.append(lon)

    points = pickle.load(open(args.outside_filename))
    for point in points:
        lat = point[0]
        lon = point[1]
        colors.append('g')
        lats.append(lat)
        lons.append(lon)

    # Draw locations as colored points (red for "inside" and green for
    # "outside)
    map.scatter(lons, lats, marker='.', c=colors, edgecolors='none', s=3, latlon=True, zorder=2, cmap=cm.plasma)
    fig = matplotlib.pyplot.gcf()
    fig.set_size_inches(18.5, 10.5)

    xs = []
    ys = []
    # Draw the eclipse path boundary as path
    for point in path.eclipse_boundary.exterior.coords:
        ys.append(point[0])
        xs.append(point[1])
    map.plot(xs, ys, latlon=True, alpha=0.5, zorder=3)

    plt.savefig(args.output, dpi=100)