web-dev-qa-db-fra.com

Réglage du format du tracé 3D

J'essaie de tracer une image 3D du fond océanique à partir des données d'un sonar parcourant une portion de 500 m sur 40 m du fond. J'utilise matplotlib/mplot3d avec Axes3D et je veux pouvoir modifier le rapport de format des axes afin que les axes x et y soient à l'échelle. Un exemple de script avec des données générées plutôt que les données réelles est:

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

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

# Generate example data.
R, Y = np.meshgrid(np.arange(0, 500, 0.5), np.arange(0, 40, 0.5))
z = 0.1 * np.abs(np.sin(R/40) * np.sin(Y/6))

# Plot the data.
surf = ax.plot_surface(R, Y, z, cmap=cm.jet, linewidth=0)
fig.colorbar(surf)

# Set viewpoint.
ax.azim = -160
ax.elev = 30

# Label axes.
ax.set_xlabel('Along track (m)')
ax.set_ylabel('Range (m)')
ax.set_zlabel('Height (m)')

# Save image.
fig.savefig('data.png')

Et l'image de sortie de ce script:

matplotlib output image

Maintenant, je voudrais le changer pour qu’un mètre dans l’axe le long de la trajectoire (x) soit égal à 1 mètre dans l’axe gamme (y) (ou peut-être un rapport différent en fonction des tailles relatives concernées). Je voudrais également définir le rapport de l'axe z, encore une fois pas nécessairement à 1: 1 en raison de la taille relative dans les données, mais l'axe est donc plus petit que le tracé actuel.

J'ai essayé de construire et d'utiliser cette branche de matplotlib , en suivant l'exemple de script dans ce message de la liste de diffusion , mais en ajoutant la ligne ax.pbaspect = [1.0, 1.0, 0.25] à mon script (après avoir désinstallé la version "standard" de matplotlib s'assurer que la version personnalisée était utilisée) ne faisait aucune différence dans l'image générée.

Edit: Ainsi, le résultat souhaité ressemblerait à l’image suivante (modifiée grossièrement avec Inkscape). Dans ce cas, je n'ai pas défini de rapport 1: 1 sur les axes x/y car cela semble ridiculement mince, mais je l'ai étalé de manière à ne pas être carré comme sur la sortie d'origine.

Desired output

pythonmatplotlibmplot3d
27
Blair

Ajoutez le code suivant avant savefig:

ax.auto_scale_xyz([0, 500], [0, 500], [0, 0.15])

enter image description here

Si vous ne voulez pas d'axe carré:

éditez la fonction get_proj dans les paquets de site\mpl_toolkits\mplot3d\axes3d.py:

xmin, xmax = np.divide(self.get_xlim3d(), self.pbaspect[0])
ymin, ymax = np.divide(self.get_ylim3d(), self.pbaspect[1])
zmin, zmax = np.divide(self.get_zlim3d(), self.pbaspect[2])

puis ajoutez une ligne pour définir pbaspect:

ax = fig.gca(projection = '3d')
ax.pbaspect = [2.0, 0.6, 0.25]

enter image description here

20
HYRY

La réponse à cette question fonctionne parfaitement pour moi. Et vous n'avez pas besoin de définir de ratio, il fait tout automatiquement.

6
user1329187

Voilà comment j'ai résolu le problème d'espace perdu:

try: 
    self.localPbAspect=self.pbaspect
    zoom_out = (self.localPbAspect[0]+self.localPbAspect[1]+self.localPbAspect[2]) 
except AttributeError: 
    self.localPbAspect=[1,1,1]
    zoom_out = 0 
xmin, xmax = self.get_xlim3d() /  self.localPbAspect[0]
ymin, ymax = self.get_ylim3d() /  self.localPbAspect[1]
zmin, zmax = self.get_zlim3d() /  self.localPbAspect[2]

# transform to uniform world coordinates 0-1.0,0-1.0,0-1.0
worldM = proj3d.world_transformation(xmin, xmax,
                                         ymin, ymax,
                                         zmin, zmax)

# look into the middle of the new coordinates
R = np.array([0.5*self.localPbAspect[0], 0.5*self.localPbAspect[1], 0.5*self.localPbAspect[2]])
xp = R[0] + np.cos(razim) * np.cos(relev) * (self.dist+zoom_out)
yp = R[1] + np.sin(razim) * np.cos(relev) * (self.dist+zoom_out)
zp = R[2] + np.sin(relev) * (self.dist+zoom_out)
E = np.array((xp, yp, zp))
0
Hexander