Faire le contour de la dispersion

En python, Si j'ai un ensemble de données

x, y, z

Je peux faire un scatter avec

import matplotlib.pyplot as plt
plt.scatter(x,y,c=z)

Comment je peux obtenir un plt.contourf(x,y,z) de l'éparpillement ?

source d'informationauteur JuanPablo

  1. 22

    Utiliser la fonction suivante pour convertir le format requis par contourf:

    from numpy import linspace, meshgrid
from matplotlib.mlab import griddata

def grid(x, y, z, resX=100, resY=100):
    "Convert 3 column data to matplotlib grid"
    xi = linspace(min(x), max(x), resX)
    yi = linspace(min(y), max(y), resY)
    Z = griddata(x, y, z, xi, yi)
    X, Y = meshgrid(xi, yi)
    return X, Y, Z

    Maintenant vous pouvez le faire:

    X, Y, Z = grid(x, y, z)
plt.contourf(X, Y, Z)

    Faire le contour de la dispersion

  2. 6

    contour attend régulièrement les données cartographiées. Vous avez donc besoin d'interpoler vos données:

    import numpy as np
from scipy.interpolate import griddata
import matplotlib.pyplot as plt
import numpy.ma as ma
from numpy.random import uniform, seed
# make up some randomly distributed data
seed(1234)
npts = 200
x = uniform(-2,2,npts)
y = uniform(-2,2,npts)
z = x*np.exp(-x**2-y**2)
# define grid.
xi = np.linspace(-2.1,2.1,100)
yi = np.linspace(-2.1,2.1,100)
# grid the data.
zi = griddata((x, y), z, (xi[None,:], yi[:,None]), method='cubic')
# contour the gridded data, plotting dots at the randomly spaced data points.
CS = plt.contour(xi,yi,zi,15,linewidths=0.5,colors='k')
CS = plt.contourf(xi,yi,zi,15,cmap=plt.cm.jet)
plt.colorbar() # draw colorbar
# plot data points.
plt.scatter(x,y,marker='o',c='b',s=5)
plt.xlim(-2,2)
plt.ylim(-2,2)
plt.title('griddata test (%d points)' % npts)
plt.show()

    Remarque que j'ai a volé sans vergogne ce code à partir de l'excellent matplotlib livre de recettes

  3. 3

    La solution dépendra de la façon dont les données sont organisées.

    De données sur grille régulière

    Si le x et y données déjà définir une grille, ils peuvent être facilement restaurées pour un quadrilatère de la grille. E. g.

    #x  y  z
 4  1  3
 6  1  8
 8  1 -9
 4  2 10
 6  2 -1
 8  2 -8
 4  3  8
 6  3 -9
 8  3  0
 4  4 -1
 6  4 -8
 8  4  8

    peut représentées par un contour à l'aide de 

    import matplotlib.pyplot as plt
import numpy as np
x,y,z = np.loadtxt("data.txt", unpack=True)
plt.contour(x.reshape(4,3), y.reshape(4,3), z.reshape(4,3))

    Arbitraire de données

    (a) Dans le cas où les données ne vit pas sur un quadrilatère de la grille, on peut interpoler les données sur une grille. Une méthode pour ce faire est fourni par matplotlib lui-même, à l'aide de matplotlib.mlab.griddata. 

    import matplotlib.mlab
xi = np.linspace(4, 8, 10)
yi = np.linspace(1, 4, 10)
zi = matplotlib.mlab.griddata(x, y, z, xi, yi, interp='linear')
plt.contour(xi, yi, zi)

    (b) Enfin, on peut tracer un contour complètement sans l'utilisation d'un quadrilatère de la grille. Cela peut être fait en utilisant tricontour.

    plt.tricontour(x,y,z)

    Un exemple en comparant les deux dernières méthodes est trouvé sur le matplotlib page.