draw a unit cube in python

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Define the vertices of the cube
vertices = np.array([[0, 0, 0], [1, 0, 0], [1, 1, 0], [0, 1, 0], 
                     [0, 0, 1], [1, 0, 1], [1, 1, 1], [0, 1, 1]])

# Define the edges that connect the vertices
edges = [(0, 1), (1, 2), (2, 3), (3, 0), 
         (4, 5), (5, 6), (6, 7), (7, 4),
         (0, 4), (1, 5), (2, 6), (3, 7)]

# Plot the vertices and edges using the plot_wireframe function
for edge in edges:
    ax.plot(vertices[edge, 0], vertices[edge, 1], vertices[edge, 2], 'b')
    
plt.show()
665 chars
22 lines