matplotlib contourf: Z-Wert unter Cursor erhalten

Question

Wenn ich etwas mit contourf plotten, sehe ich am unteren Rand des Diagrammfensters die aktuellen x- und y-Werte unter dem Mauszeiger. Gibt es eine Möglichkeit, auch den z-Wert zu sehen?

Hier ein Beispiel contourf:

import matplotlib.pyplot as plt 
import numpy as hp 
plt.contourf(np.arange(16).reshape(-1,4)) 

Answer 1

Die documentation example zeigt, wie Sie z-Wert von Etiketten in Ihrem Grundstück

Script einfügen können: http://matplotlib.sourceforge.net/mpl_examples/pylab_examples/contour_demo.py

Im Grunde ist es

plt.figure() 
CS = plt.contour(X, Y, Z) 
plt.clabel(CS, inline=1, fontsize=10) 
plt.title('Simplest default with labels') 

Answer 2

Der Text, der die Positio zeigt n des Cursors wird von ax.format_coord generiert. Sie können die Methode überschreiben, um auch einen Z-Wert anzuzeigen. Zum Beispiel

import matplotlib.pyplot as plt 
import numpy as np 
import scipy.interpolate as si 
data = np.arange(16).reshape(-1, 4) 
X, Y = np.mgrid[:data.shape[0], :data.shape[1]] 
cs = plt.contourf(X, Y, data) 


def fmt(x, y): 
    z = np.take(si.interp2d(X, Y, data)(x, y), 0) 
    return 'x={x:.5f} y={y:.5f} z={z:.5f}'.format(x=x, y=y, z=z) 


plt.gca().format_coord = fmt 
plt.show() 

Answer 3

Nur eine Variante von wilywampa ‚s Antwort. Wenn Sie bereits ein vorberechnetes Gitter aus interpolierten Konturwerten haben, weil Ihre Daten spärlich sind oder wenn Sie eine riesige Datenmatrix haben, könnte dies für Sie geeignet sein.

import matplotlib.pyplot as plt 
import numpy as np 

resolution = 100 
Z = np.arange(resolution**2).reshape(-1, resolution) 
X, Y = np.mgrid[:Z.shape[0], :Z.shape[1]] 
cs = plt.contourf(X, Y, Z) 

Xflat, Yflat, Zflat = X.flatten(), Y.flatten(), Z.flatten() 
def fmt(x, y): 
    # get closest point with known data 
    dist = np.linalg.norm(np.vstack([Xflat - x, Yflat - y]), axis=0) 
    idx = np.argmin(dist) 
    z = Zflat[idx] 
    return 'x={x:.5f} y={y:.5f} z={z:.5f}'.format(x=x, y=y, z=z) 

plt.colorbar() 
plt.gca().format_coord = fmt 
plt.show() 

Ex: