Построение 3D-полигонов в python-matplotlib

Я был неудачным веб-браузером для решения для следующего простого вопроса:


Как нарисовать 3D-многоугольник (например, заполненный прямоугольник или треугольник), используя значения вершин? Я пробовал много идей, но все не удалось, см.:

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()

Я заранее ценю любую идею/комментарий.

Обновления, основанные на принятом ответе:

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()

Вот результат: enter image description here

Ответ 1

Я думаю, что вы почти получили это. Это то, что вы хотите?

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 Вас также может заинтересовать art3d.pathpatch_2d_to_3d.

Ответ 2

Вышеупомянутое решение предназначено для Python 2 и выдает ошибку 'TypeError: объект типа' zip 'не имеет len()' при запуске с python 3.

См. Раздел " Создание 3D-полигонов в Python 3" для обсуждения обновления этого до Python 3.

Вот некоторый рабочий код оттуда:

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()