So erstellen Sie ein 3D-Diagramm in Matplotlib aus Daten z = f (x, y) aus der Datei lesen

Question

Ich muss ein 3D-Diagramm in Matplotlib erstellen. Ich lese Z-Werte aus einer Textdatei, die als 1D-Array formatiert ist. Diese Daten repräsentieren den Wert für jeden Punkt auf einem 50x50-Quadrat. Also sind die ersten 50 Werte für Punkte (0,0), (1,0) .. (49,0), die nächsten 50 sind für (0,1), (1,1) und so weiter. Bisher habe ich folgenden Code geschrieben:

import matplotlib.pyplot as plt 
    import numpy as np 

    fp=open(path,"r") 
    a=fp.read() 
    buffer="" 
    data=[] 
    for i in a: 
     if i != ' ': 
      buffer=buffer+i 
     else: 
      data.append(float(buffer)) 
      buffer="" 
    fp.close() 


    values=data[50*50:50*50*2] 

    x=np.linspace(0,50,50) 
    y=np.linspace(0,50,50) 

    fig = plt.figure() 
    ax = fig.gca(projection='3d') 

    ax.plot_trisurf(x, y, values) 

    plt.show() 

Aber natürlich funktioniert es nicht. Ich suchte im Internet nach Hinweisen, aber ohne Erfolg. Kann mir jemand zeigen, wie ich es machen soll? Ich wäre mehr als dankbar.

Answer 1

Zuerst haben Sie vergessen, Axes3D zu importieren: import matplotlib.pyplot as plt.

Zweitens haben Sie Code merkwürdig sinnlose Zeile: values[50*50*nb:50*50*(nb+1)] und Einzug ist falsch in Ihrem Code.

Drittens plotten regelmäßig gerasterten Daten, die Sie nicht plot_trisurf benötigen, verwenden plot_surface:

import matplotlib.pyplot as plt 
import numpy as np 
from mpl_toolkits.mplot3d import Axes3D 

x=np.linspace(0,50,50) 
y=np.linspace(0,50,50) 
X, Y = np.meshgrid(x, y) 
values = np.sqrt(X**2 + Y**2) 

fig = plt.figure() 
ax = fig.gca(projection='3d') 
ax.plot_surface(x, y, values) 
plt.show() 

Geben Sie für Beispiele: http://matplotlib.org/examples/mplot3d/surface3d_demo.html