Hi, im trying to create a heatmap/colourmap/contour style plot similar to
http://www.idlcoyote.com/cg_tips/outcontourbar.png but I cant seem to get it
working correctly. The code takes a number of parameters on the command line
because it is passed by an external PHP script. Essentially its a list of
lat/lon/values which I want to interpolate and plot but I cant get anything
like the image above, I cant get the vmin/vmax to work for the colorbar and
I dont know why the whole plot is squared off rather than fading to white if
you see what I mean
import os
import tempfile
import math
os.environ['MPLCONFIGDIR'] = tempfile.mkdtemp()
import argparse
import numpy as np
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import scipy.interpolate
width = 800
height = 600
lat_min = []
lon_min = []
lat_max = []
lon_max = []
# assumes lat is y, lon is x, and image is 800x600
def latToXY(lat):
global width, height, lat_min, lat_max
y = ((lat - lat_min) / (lat_max - lat_min)) * height
#print y
return y
def lonToXY(lon):
global width, height, lon_min, lon_max
lon = math.fabs(lon)
x = ((lon - lon_min) / (lon_max - lon_min)) * width
#print x
return x
def scipy_idw(x, y, z, xi, yi):
#interp = scipy.interpolate.Rbf(x, y, z, function='linear')
interp = scipy.interpolate.Rbf(x, y, z)
return interp(xi, yi)
def plot(x,y,z,grid,legend_min,legend_max,filename):
plt.figure()
fig = plt.imshow(grid, vmin=legend_min, vmax=legend_max, extent=[0,
1024, 0, 768])
fig.axes.get_xaxis().set_visible(False) # hide axis labels
fig.axes.get_yaxis().set_visible(False)
#plt.hold(True)
plt.scatter(x,y,c=z)
plt.colorbar()
plt.savefig(filename)
# grab all floats from command line
parser = argparse.ArgumentParser()
parser.add_argument('--l1', type=str)
parser.add_argument('--l2', type=str)
parser.add_argument('--l3', type=str)
parser.add_argument('--min', type=str)
parser.add_argument('--max', type=str)
parser.add_argument('--filename', type=str)
args = parser.parse_args()
# create a list by splitting at the comma
l1_list = args.l1.split(',') # ['1','2','3','4']
l2_list = args.l2.split(',')
l3_list = args.l3.split(',')
legend_min = float(args.min)
legend_max = float(args.max)
filename = args.filename
# convert string list to list of floats
for i in range(len(l1_list)):
l1_list[i] = float(l1_list[i])
l2_list[i] = float(l2_list[i])
l3_list[i] = float(l3_list[i])
lat_min = min( math.fabs(yy) for yy in l2_list )
lat_max = max( math.fabs(yy) for yy in l2_list )
lon_min = min( math.fabs(xx) for xx in l1_list )
lon_max = max( math.fabs(xx) for xx in l1_list )
# convert list of floats to x,y
for i in range(len(l1_list)):
l1_list[i] = lonToXY(l1_list[i])
l2_list[i] = latToXY(l2_list[i])
# convert list to numpy array
x = np.array(l1_list)
y = np.array(l2_list)
z = np.array(l3_list)
#print x
#print y
#print z
nx, ny = 50, 50
xi, yi = np.linspace(x.min(), x.max(), nx), np.linspace(y.min(), y.max(),
ny)
xi, yi = np.meshgrid(xi, yi)
xi, yi = xi.flatten(), yi.flatten()
grid2 = scipy_idw(x,y,z,xi,yi)
grid2 = grid2.reshape((ny, nx))
plot(x,y,z,grid2,legend_min,legend_max,filename)
I call the script with the following parameters:
plot.py --l1=-1.8791363,-1.8786206,-1.8796862,-1.878171
--l2=57.458459,57.458153,57.458495,57.458036 --l3=42.3,37.8,43.5,47.7
--min=0 --max=100 --filename=/tmp/plot.png
Any help is greatly appreciated
Thanks
···
--
View this message in context: http://matplotlib.1069221.n5.nabble.com/matplolib-imshow-contour-heatmap-tp43078.html
Sent from the matplotlib - users mailing list archive at Nabble.com.
