Pirämide. Simple Python

"""Crear el gráfico de una pirámide"""

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.art3d import Poly3DCollection

# Definir las coordenadas de los vértices de la pirámide
vertices = np.array([[0, 0, 0],
                     [1, 0, 0],
                     [1, 1, 0],
                     [0, 1, 0],
                     [0.5, 0.5, 1]])

# Definir las caras de la pirámide
caras = [[vertices[0], vertices[1], vertices[4]],
         [vertices[0], vertices[3], vertices[4]],
         [vertices[2], vertices[1], vertices[4]],
         [vertices[2], vertices[3], vertices[4]],
         [vertices[0], vertices[1], vertices[2], vertices[3]]]

# Crear la figura y los ejes 3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Dibujar las caras de la pirámide
ax.add_collection3d(Poly3DCollection(caras, facecolors='lightgrey', linewidths=1, edgecolors='black', alpha=0.5))

# Configuración del aspecto
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('Pirámide Tridimensional')
ax.set_xlim([0, 1])
ax.set_ylim([0, 1])
ax.set_zlim([0, 1])
ax.view_init(elev=30, azim=45)  # Ajustar la vista, observador esta fuera de la pirámide

# Mostrar el gráfico
plt.show()