Die Generierung einer heat-map mit 3D-Daten in matplotlib

Habe ich eine Funktion returnValuesAtTime dass gibt drei Listen-x_vals,y_vals und swe_vals. Alle drei Listen sind von gleicher Länge und jedes element in swe_vals entspricht einer x-value aus x_vals und ein y-value aus y_vals. Ich möchte erzeugen eine heatmap mit einer Legende, verwendet die (x,y) Koordinaten und die swe_vals wie die Intensität.

Schrieb ich den folgenden code:

def plotValuesAtTimeMap(t):
    x_vals,y_vals,swe_vals=returnValuesAtTime(t)
    x_points=len(x_vals)
    y_points=len(y_vals)
    xx=np.linspace(x_vals[0],x_vals[-1],x_points)
    yy=np.linspace(y_vals[0],y_vals[-1],y_points)
    fig,ax=plt.subplots()
    im=ax.pcolormesh(xx,yy,z)
    fig.colorbar(im)
    ax.axis('tight')
    plt.show()

Einmal die drei Listen, die mit Hilfe returnValuesAtTime(t) ich nehme die Längen der x_vals und y_vals. Dann verwende ich diese zu generieren, die den Abstand für die x-und y-Richtungen, mit den Grenzen, die ersten und letzten Elemente der x_vals und y_vals. Ich versuche dann zu generieren, die colormesh. Aber dies gibt mir eine leere colormesh ohne Werte.

Was könnte schief gehen? Wird mit Hilfe eines 3D - numpy array statt der drei Listen, das problem zu lösen?

Den ersten 10 Elemente der einzelnen Listen sind:

x_vals[0:10]

[439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893,
 439936.86573189893]

y_vals[0:10]

[4514018.2797159087,
 4513898.2797159087,
 4513778.2797159087,
 4513658.2797159087,
 4513538.2797159087,
 4513418.2797159087,
 4513298.2797159087,
 4513178.2797159087,
 4513058.2797159087,
 4512938.2797159087]

swe_vals[0:10]

[2.7520323,
 2.7456229,
 2.7456021,
 2.745455,
 2.7478349,
 2.7445269,
 2.7484877,
 2.7524617,
 2.75491,
 2.7509627]

Bearbeiten:

Ich habe eine scatter-plot zeigt die (x,y) grid-Bereich unter:

Die Generierung einer heat-map mit 3D-Daten in matplotlib

Den x-und y-Werte in Listen, 6804 Länge. Jedes (x,y) - Punkt hat eine entsprechende z-Wert in einer separaten Liste, die auch 6804 Länge. Um zu verdeutlichen, was ich hoffe zu erreichen, ich möchte erzeugen eine heatmap wie Grundstück mit der Größe der z-Achse, dargestellt durch die Farbe der einzelnen raster auf dem Grundstück. So etwas wie, was ist unten dargestellt:

Die Generierung einer heat-map mit 3D-Daten in matplotlib

In der Beispiel-plot, alle z-Werte sind die gleichen. Also die Farbe ist die gleiche über die gesamte grid-Raum.

Bearbeitet mit neuen resultierenden plot (basierend auf den CT-Zhu Vorschläge):

Die Generierung einer heat-map mit 3D-Daten in matplotlib

Ist sicher davon ausgehen, dass Ihre x_vals und y_vals nicht gleichmäßig verteilt, wie 1,2,3,4,....? Meistens wahrscheinlich ja, wie Sie haben eine Reihe von xx=np.linspace(x_vals[0],x_vals[-1],x_points)?
Sie sind gleichmäßig verteilt, wobei die ersten drei Elemente der x_vals wird [ 439936.8657319, 440056.8657319, 440176.8657319] und die ersten drei Elemente der y_vals wird [ 4514018.27971591, 4513898.27971591, 4513778.27971591]. Jedes element in x_vals wird 120.0 mehr als im Vorjahr, und jedes element in y_vals wird 120.0 weniger als die Vorherige.
Ich sehe, wie ein kommen aus meshgrid

InformationsquelleAutor small_world | 2014-04-04

5

Sieht es aus, wenn umzugestalten x, y, z zu der quadratischen matrix, die Sie tun können, eine contourf plot:
```
In [7]:X
Out[7]:
array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])

In [8]:Y
Out[8]:
array([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
       [1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
       [2, 2, 2, 2, 2, 2, 2, 2, 2, 2],
       [3, 3, 3, 3, 3, 3, 3, 3, 3, 3],
       [4, 4, 4, 4, 4, 4, 4, 4, 4, 4],
       [5, 5, 5, 5, 5, 5, 5, 5, 5, 5],
       [6, 6, 6, 6, 6, 6, 6, 6, 6, 6],
       [7, 7, 7, 7, 7, 7, 7, 7, 7, 7],
       [8, 8, 8, 8, 8, 8, 8, 8, 8, 8],
       [9, 9, 9, 9, 9, 9, 9, 9, 9, 9]])

plt.contourf(X,Y,np.random.random((10,10))) #reshape Z too!
plt.colorbar()
```
- Gibt es einen bestimmten Unterschied zwischen einer matrix und einem array? Ich habe eine andere Funktion zurückgibt, die ein 3D-numpy-array enthält die x-Koordinate, y-Koordinate und der z-Wert (swe_vals). Die Funktion ist: def createArray(t): x_vals,y_vals,swe_vals=returnValuesAtTime(t) swe_array=np.array([x_vals,y_vals,swe_vals],np.float32) return swe_array . Gibt es eine Möglichkeit, ich kann wandeln diese in matrix und erzeugen den Graphen?
- Jeder meiner Listen (x,y und z) 6804-Werte. Ist es möglich, zu konvertieren, das zu einer quadratischen matrix? Ich habe versucht mehrere Methoden, aber keiner von Ihnen scheint zu arbeiten.
- Wenn Sie einen scatter-plot der x und y ist, tut es scheint eine gleichmäßig verteilte Quadrate? 6804 ist ein wenig seltsam und nicht zu einer quadratischen matrix, 82*83=8606
- Ich habe einige Beispiel-plots in meiner Frage bearbeitet. 6804 nicht zu einer quadratischen matrix.
- Es sieht aus wie Ihre raster ist in der Tat 84*81, können Sie es überprüfen, indem len(set(x_vals))? Wenn das Gitter 84*81, können Sie die heatmap wie beschrieben, aber verändern Sie Ihre Daten in (84,81) dann eher eine quadratische matrix.
- Danke, das hat funktioniert! Aber die y-Achse scheint nicht zu wurden in vollem Umfang gezeichnet. Der maximale Wert auf meine y-Achse sollte etwa 4514018.28, aber der plot zeigt nur bis 10000, und zeigt +4.50 E-6 an der Spitze. Wissen Sie, warum dies passiert ist? Ich habe aktualisiert, die neue Handlung in den Bearbeitungen auf meine Frage.
- Das sollte das normale Verhalten, wenn die zahlen groß sind. Aber ich denke, es bedeutet, dass Ihre größten Y-Werte sind irgendwo um 10000*4.5*1000000.
InformationsquelleAutor CT Zhu

Schreibe einen Kommentar

Du musst angemeldet sein, um einen Kommentar abzugeben.