3dpyramid.py

# http://matplotlib.1069221.n5.nabble.com/attachment/4748/0/3dpyramid.py

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

fig=plt.figure()
ax = Axes3D(fig)

# Face 1
x1 = np.array([[0, 0, 1, 1, 0],
               [0, 0, 0, 0, 0]])
y1 = np.array([[0, 1, 1, 0, 0],
               [0, 0, 0, 0, 0]])
z1 = np.array([[0, 0, 0, 0, 0],
               [0, 0, 0, 0, 0]])

# Face 2
x2 = np.array([[0, 0, 0.5, 0],
               [0, 0, 0, 0]])
y2 = np.array([[0, 1, 0.5, 0],
               [0, 0, 0, 0]])
z2 = np.array([[0, 0, 1, 0],
               [0, 0, 0, 0]])
# Face 3
x3 = np.array([[1, 1, 0.5, 1],
               [1, 1, 1, 1]])
y3 = np.array([[0, 1, 0.5, 0],
               [0, 0, 0, 0]])
z3 = np.array([[0, 0, 1, 0],
               [0, 0, 0, 0]])
# Face 4
x4 = np.array([[0, 1, 0.5, 0],
               [0, 0, 0, 0]])
y4 = np.array([[1, 1, 0.5, 1],
               [1, 1, 1, 1]])
z4 = np.array([[0, 0, 1, 0],
               [0, 0, 0, 0]])
# Face 5
x5 = np.array([[1, 0, 0.5, 1],
               [1, 1, 1, 1]])
y5 = np.array([[0, 0, 0.5, 0],
               [0, 0, 0, 0]])
z5 = np.array([[0, 0, 1, 0],
               [0, 0, 0, 0]])

ax.plot_surface(x1,y1,z1)
ax.plot_surface(x2,y2,z2)
ax.plot_surface(x3,y3,z3)
ax.plot_surface(x4,y4,z4)
ax.plot_surface(x5,y5,z5)

plt.show()