Python : 2d contour plot from 3 lists : x, y and rho?

by

You need to interpolate your rho values. There’s no one way to do this, and the “best” method depends entirely on the a-priori information you should be incorporating into the interpolation.

Before I go into a rant on “black-box” interpolation methods, though, a radial basis function (e.g. a “thin-plate-spline” is a particular type of radial basis function) is often a good choice. If you have millions of points, this implementation will be inefficient, but as a starting point:

import numpy as np
import matplotlib.pyplot as plt
import scipy.interpolate

# Generate data:
x, y, z = 10 * np.random.random((3,10))

# Set up a regular grid of interpolation points
xi, yi = np.linspace(x.min(), x.max(), 100), np.linspace(y.min(), y.max(), 100)
xi, yi = np.meshgrid(xi, yi)

# Interpolate
rbf = scipy.interpolate.Rbf(x, y, z, function='linear')
zi = rbf(xi, yi)

plt.imshow(zi, vmin=z.min(), vmax=z.max(), origin='lower',
           extent=[x.min(), x.max(), y.min(), y.max()])
plt.scatter(x, y, c=z)
plt.colorbar()
plt.show()

enter image description here

More Related Contents:

  1. Make contour of scatter
  2. Why does pyplot.contour() require Z to be a 2D array?
  3. matplotlib – extracting data from contour lines
  4. Increase the speed of redrawing contour plot in matplotlib
  5. Python: find contour lines from matplotlib.pyplot.contour()
  6. Matplotlib scatterplot; color as a function of a third variable
  7. Set Colorbar Range in matplotlib
  8. Matplotlib make tick labels font size smaller
  9. matplotlib does not show my drawings although I call pyplot.show()
  10. Seaborn Barplot – Displaying Values
  11. How to make a 4d plot with matplotlib using arbitrary data
  12. Matplotlib log scale tick label number formatting
  13. Is it possible to get color gradients under curve in matplotlib?
  14. Plotting a decision boundary separating 2 classes using Matplotlib’s pyplot
  15. How to change legend size with matplotlib.pyplot
  16. Changing position of vertical (z) axis of 3D plot (Matplotlib)?
  17. How to save a Seaborn plot into a file
  18. Putting arrowheads on vectors in matplotlib’s 3d plot
  19. Animate points with labels with matplotlib
  20. Matplotlib – logarithmic scale, but require non-logarithmic labels
  21. How to crop the internal area of a contour?
  22. Error in installing Matplotlib : fatal error C1083
  23. ImportError: Cannot load backend ‘TkAgg’ which requires the ‘tk’ interactive framework, as ‘headless’ is currently running
  24. Python making combined bar and line plot with secondary y-axis
  25. Cancelling the offset on an axis [duplicate]
  26. Plot inline or a separate window using Matplotlib in Spyder IDE
  27. import matplotlib.pyplot hangs
  28. Is it possible to add hatches to each individual bar in seaborn.barplot?
  29. Seaborn multiple barplots
  30. pandas scatter plotting datetime

Leave a Comment