def Plot_Station_Data(): lons = load('%s/lons_good_stations' % workspace) lats = load('%s/lats_good_stations' % workspace) arid = [] for ibasin in xrange(0, 10): arid.append( load('%s/aridity_station_%s' % (workspace, basinlongs[ibasin]))) arid = array(list(itertools.chain(*arid))) index_DI = msc_groupby_DI(arid) # print [len(index_DI[i]) for i in xrange(0, 4)] # print mean(arid), median(arid) # print [median(arid[index_DI[i]]) for i in xrange(0, 4)] colors = r_[linspace(0, 1, 4), linspace(0, 1, 4)] mymap = plt.get_cmap("coolwarm") my_colors = mymap(colors) arid_class = ['humid', 'sub-humid', 'semi-arid', 'arid'] [ Plotting.Mapshow_Aridity(lons[index_DI[i]], lats[index_DI[i]], 50, 0.8, 0, 1000, my_colors[i], arid_class[i]) for i in xrange(0, 4) ] # plt.show() fig = plt.gcf() fig.set_size_inches(6, 5) plt.show() # savefig('%s/map_aridity_good_station.tif' %figdir, dpi=300) return