1

I'm trying to set up a contour plot but the figure ends up looking like this:

fig1.

I think it's because the y axis goes from around 10 to 80 units and the x only 4 to 9 units and the mesh is square so each unit is scaled the same on the x and y axis. How can I change the scaling of the x axis to make it a usable image?

import numpy as np
import matplotlib.pyplot as plt
import time as time
from scipy.interpolate import griddata

x = np.array([3.98, 3.98, 3.99, 3.98, 6.02, 6.05, 6.03, 6.04, 10.01, 10.07, 10.09, 10.1])
y = np.array([79.5, 40, 20.07, 10.04, 79.67, 40.08, 20.23, 10.01, 79.42, 39.63, 20.14, 10.11])
z = np.array([38.82697111, 34.61715402, 22.46052757, 13.51888155, 48.52466738, 42.1140194, 28.37788811, 16.48901657, 55.40461783, 46.88792485, 33.66091083, 20.9200696])

extent = (min(x), max(x), min(y), max(y))
xs,ys = np.mgrid[extent[0]:extent[1]:30j, extent[2]:extent[3]:30j]
resampled = griddata((x, y), z, (xs, ys))
plt.figure(figsize=(8,8))
plt.imshow(resampled.T, extent=(min(x), max(x), max(y), min(y)),interpolation='bicubic')
plt.contour(resampled.T, extent=(min(x), max(x), max(y), min(y)),interpolation='bicubic',origin='upper')
plt.hold(True)
plt.scatter(x,y,c=z)
plt.gca().invert_yaxis()
plt.colorbar()
plt.show()
tmdavison
  • 64,360
  • 12
  • 187
  • 165
L.Bouwmeester
  • 11
  • 2

1 Answers1

1

Simply set the aspect keyword, either manually:

plt.imshow(resampled.T, extent=(min(x), max(x), max(y), min(y)),interpolation='bicubic', aspect=0.1)

or by using the auto option:

plt.imshow(resampled.T, extent=(min(x), max(x), max(y), min(y)),interpolation='bicubic', aspect='auto')

Another option is to use pcolormesh()

Bart
  • 9,825
  • 5
  • 47
  • 73