from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import matplotlib.animation as animation

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

# Grab some example data.
X, Y, Z = axes3d.get_test_data(0.05)

# Plot a basic wireframe.
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)

# Set the axis labels
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')

# Define an update function to pass to FuncAnimation
def update_view(frame):
    # Normalize angle to the range [-180, 180] for display
    angle = (frame + 180) % 360 - 180

    # Cycle through rotating the elevation, azimuth, roll, then all
    elev = azim = roll = 0
    if frame <= 360:
        elev = angle
    elif frame <= 360*2:
        azim = angle
    elif frame <= 360*3:
        roll = angle
    else:
        elev = azim = roll = angle
    
    # Update the axis view and title
    ax.view_init(elev, azim, roll)
    plt.title('Elevation: %d°, Azimuth: %d°, Roll: %d°' % (elev, azim, roll))

# Plot the animation
anim = animation.FuncAnimation(fig, update_view, frames=360*4+1, interval=15)
anim.save("3d_rotations.mp4")