Comment faire des tracés 3D en Python?

Question

Ceci est la version Matlab du code de traçage 3D: EDIT: Voici le code actuel:

 plt.figure(2) 
     fig_b = Axes3D(fig2) 
     xx2 = np.arange(0, L+h_grid*L, h_grid*L) 
     yy2 = np.arange(-b, b+h_grid*b, h_grid*b) 
     X, Y = np.meshgrid(xx2, yy2) 
     W = np.zeros((41,21), float) 
     mx = len(xx2)*len(yy2) 
     X = np.reshape(X, (1, mx)) 
     Y = np.reshape(Y, (1, mx)) 
     W = np.reshape(W, (1, mx)) 
     for j in range(0, mx): 
      W[0][j] = np.sin(np.pi*X[0][j]/L) 
     surf = fig_b.plot_surface(X, Y, W, rstride=1, cstride=1, cmap=cm.jet, linewidth=0, antialiased=False) # This is the line number 168 
     plt.xlabel('x') 
     plt.ylabel('y') 

Ceci est le message d'erreur que je reçois:

Traceback (most recent call last): 
    File "nonhomog.py", line 247, in <module> 
    main() 
    File "nonhomog.py", line 245, in main 
    nonhomog(nu) 
    File "nonhomog.py", line 168, in nonhomog 
    surf = fig_b.plot_surface(X, Y, W, rstride=1, cstride=1, cmap=cm.jet, linewidth=0, antialiased=False) 
    File "/usr/lib/pymodules/python2.6/mpl_toolkits/mplot3d/axes3d.py", line 618, in plot_surface 
    polyc = art3d.Poly3DCollection(polys, *args, **kwargs) 
    File "/usr/lib/pymodules/python2.6/mpl_toolkits/mplot3d/art3d.py", line 290, in __init__ 
    PolyCollection.__init__(self, verts, *args, **kwargs) 
    File "/usr/lib/pymodules/python2.6/matplotlib/collections.py", line 668, in __init__ 
    self.set_verts(verts, closed) 
    File "/usr/lib/pymodules/python2.6/mpl_toolkits/mplot3d/art3d.py", line 312, in set_verts 
    self.get_vector(verts) 
    File "/usr/lib/pymodules/python2.6/mpl_toolkits/mplot3d/art3d.py", line 305, in get_vector 
    xs, ys, zs = zip(*points) 
ValueError: need more than 0 values to unpack 

Answer 1

La façon dont je fais actuellement dans mon code est:

# [ (x1, y1, z1), (x2, y2, z2), ... (xN, yN, zN) ] 
all_vals = ... 
# (x1, x2, ... xN) , (y1, y2, ... yN) , (z1, z2, ... zN) 
all_xvals, all_yvals, all_zvals = zip(*all_vals) 
fig = plt.figure() 
ax = Axes3D(fig) 
X, Y = np.meshgrid(xvals, yvals) 
# This is the part you want: 
Z1 = np.zeros(X.shape, float) 
for (x, y, z) in all_vals: 
    x = find_in_sorted_list(x, xvals) 
    y = find_in_sorted_list(y, yvals) 
    Z1[y,x] = z  
surf = ax.plot_surface(X, Y, Z1, rstride=1, cstride=1, cmap=cm.jet, 
     linewidth=0, antialiased=False) 
plt.xlabel('Blur standard deviation') 
plt.ylabel('JPEG quality') 
ax.w_zaxis.set_major_locator(LinearLocator(10)) 
ax.w_zaxis.set_major_formatter(FormatStrFormatter('%.03f'))  
fig.colorbar(surf, shrink=0.5, aspect=5)  
plt.show() 

Cela me donne une intrigue qui ressemble à ceci:

Je l'ai sauvé dans un fichier, mais quand vous appelez plt.show(), vous obtenez une fenêtre interactive où vous pouvez changer le point de vue à tout ce que vous voulez.

Answer 2

Qu'est-ce qui ne va pas? Vous essayez de faire un non-nombre négatif. En d'autres termes: AxesSubplot (quel qu'il soit) n'implémente pas l'opérateur unaire. Donc, ce code ne peut pas raisonnablement être "ce que vous avez fait", car vous ne définissez même pas b dans ce code, mais il existe et est de type personnalisé appelé AxesSubplot. Si vous expliquez ce qu'est AxesSubplot, cela vous aidera. Essayez d'inclure du code qui démontre le problème, si possible. Comme l'indique DSM, vous remplacez votre variable b. Le problème est que vous êtes coincé en "mode maths" et utilisez des noms de variables non descriptives comme "a", "b" et "M". Utilisez plutôt des noms descriptifs plus longs.

Au lieu de:

a = fig.add_subplot(2,2,i) 
b = fig2.add_subplot(2,2,i) 

faire:

x_subplot = fig.add_subplot(2,2,i) 
y_subplot = fig2.add_subplot(2,2,i) 

Ou quelque chose comme ça (je ne suis pas sûr de ce que la variable effectivement est, de sorte que est juste un exemple). Après avoir configuré la grille de maillage pour X et Y, vous devez créer une grille pour les valeurs de Z.