An update upon the previous.
In this instance I’ve defined the function at one location within the
code; this is useful when you want to plot an arbitrary function.
# projections_4b.py
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.animation import FuncAnimation
import matplotlib.animation as animation
plt.style.use('dark_background')
# Define a custom function to generate test data for the surface plot
def generate_test_data(t):
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)
s = np.mod(t,2)
Z = np.sin(np.sqrt(X**2 + Y**2 )*np.pi*s+0.00001)/(np.sqrt(X**2 + Y**2 )*np.pi*s+0.00001)
return X, Y, Z
# Create a 3D figure
fig = plt.figure()
axs = fig.add_subplot(111, projection='3d')
# Initialize an empty surface plot
ae=np.empty([2,2])
surf = axs.plot_surface(ae, ae, ae, cmap='viridis')
# Set the perspective projection
axs.set_proj_type('persp', focal_length=0.2) # FOV = 157.4 deg
# Set the background color to black
axs.set_facecolor('black')
# Set the title
axs.set_title("'persp'\nfocal_length = 0.2", fontsize=10)
# Function to update the plot for each frame
def update_plot(frame):
global surf
t = frame * 0.1 # Adjust the time parameter for animation
X, Y, Z = generate_test_data(t)
surf.remove()
surf = axs.plot_surface(X, Y, Z, cmap='viridis')
return surf,
# Create an animation
ani = animation.FuncAnimation(fig, update_plot, frames=100, interval=100)
plt.show()