web-dev-qa-db-fra.com

Tracer des polygones 3D en python-matplotlib

Je n'ai pas réussi à naviguer sur le Web pour trouver une solution à la question simple suivante:

Comment dessiner un polygone 3D (disons un rectangle ou un triangle rempli) en utilisant des valeurs de sommets? J'ai essayé de nombreuses idées mais toutes ont échoué, voir:

from mpl_toolkits.mplot3d import Axes3D
from matplotlib.collections import PolyCollection
import matplotlib.pyplot as plt
fig = plt.figure()
ax = Axes3D(fig)
x = [0,1,1,0]
y = [0,0,1,1]
z = [0,1,0,1]
verts = [Zip(x, y,z)]
ax.add_collection3d(PolyCollection(verts),zs=z)
plt.show()

J'apprécie à l'avance toute idée/commentaire.

Mises à jour basées sur la réponse acceptée:

import mpl_toolkits.mplot3d as a3
import matplotlib.colors as colors
import pylab as pl
import scipy as sp

ax = a3.Axes3D(pl.figure())
for i in range(10000):
    vtx = sp.Rand(3,3)
    tri = a3.art3d.Poly3DCollection([vtx])
    tri.set_color(colors.rgb2hex(sp.Rand(3)))
    tri.set_edgecolor('k')
    ax.add_collection3d(tri)
pl.show()

Voici le résultat: enter image description here

pythonmatplotlibplot
54
Developer

Je pense que vous l'avez presque compris. c'est ce que tu veux?

from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
import matplotlib.pyplot as plt
fig = plt.figure()
ax = Axes3D(fig)
x = [0,1,1,0]
y = [0,0,1,1]
z = [0,1,0,1]
verts = [list(Zip(x,y,z))]
ax.add_collection3d(Poly3DCollection(verts))
plt.show()

alt text Vous pourriez également être intéressé par art3d.pathpatch_2d_to_3d.

50
DSM

La solution ci-dessus est pour Python 2, et donne une erreur 'TypeError: l'objet de type' Zip 'n'a pas de len ()' lorsqu'il est exécuté avec python 3 .

Voir Plotting 3D Polygons in Python pour une discussion sur la mise à jour de ceci vers Python 3.

Voici un code de travail à partir de là:

from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
import matplotlib.pyplot as plt

fig = plt.figure()
ax = Axes3D(fig)
x = [0, 1, 1, 0]
y = [0, 0, 1, 1]
z = [0, 1, 0, 1]
verts = [list(Zip(x, y, z))]
print(verts)
ax.add_collection3d(Poly3DCollection(verts), zs='z')
plt.show()
4
Alex Kritchevsky