四变量函数可视化咨询：Python/Matlab时空动画实现建议

We'll use matplotlib for 3D plotting and animation, paired with numpy for numerical computations.

Step-by-Step Code #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# Define your 4-variable function (adjust this to your actual function)
def four_var_func(x, y, z, t):
    return np.sin(x + t) * np.cos(y - t) * np.tanh(z)

# Create 3D grid of points
x = np.linspace(-np.pi, np.pi, 20)
y = np.linspace(-np.pi, np.pi, 20)
z = np.linspace(-2, 2, 20)
X, Y, Z = np.meshgrid(x, y, z)
# Flatten grids to 1D arrays for scatter plotting
x_flat = X.flatten()
y_flat = Y.flatten()
z_flat = Z.flatten()

# Initialize plot
fig = plt.figure(figsize=(10, 8))
ax = fig.add_subplot(projection='3d')
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('4-Variable Function Animation (t as color)')

# Initial scatter plot (t=0)
initial_vals = four_var_func(x_flat, y_flat, z_flat, 0)
scatter = ax.scatter(x_flat, y_flat, z_flat, c=initial_vals, cmap='viridis', s=10)
plt.colorbar(scatter, label='Function Value')

# Animation update function
def update(frame):
    t = frame * 0.1  # Adjust time step to control animation speed
    current_vals = four_var_func(x_flat, y_flat, z_flat, t)
    scatter.set_array(current_vals)
    ax.set_title(f'4-Variable Function Animation (t = {t:.1f})')
    return scatter,

# Create and run animation
ani = FuncAnimation(fig, update, frames=100, interval=50, blit=True)

# Optional: Save animation to file (requires ffmpeg)
# ani.save('4var_animation.mp4', writer='ffmpeg', fps=20)

plt.show()

Key Notes #

• Adjust grid resolution (the 20 in linspace) to balance visual smoothness and performance—more points mean richer detail but slower playback.
• Swap cmap to use different color scales (e.g., plasma, coolwarm, inferno).
• Modify the t calculation in update to change time progression speed or range.

Matlab's scatter3 and built-in animation tools simplify this workflow.

Step-by-Step Code #

% Define your 4-variable function (adjust to your actual function)
function val = four_var_func(x, y, z, t)
    val = sin(x + t) .* cos(y - t) .* tanh(z);
end

% Create 3D grid of points
x = linspace(-pi, pi, 20);
y = linspace(-pi, pi, 20);
z = linspace(-2, 2, 20);
[X, Y, Z] = meshgrid(x, y, z);
% Flatten grids to column vectors
x_flat = X(:);
y_flat = Y(:);
z_flat = Z(:);

% Initialize plot
figure('Position', [100, 100, 800, 600]);
ax = axes('Projection', '3d');
xlabel('X'); ylabel('Y'); zlabel('Z');
title('4-Variable Function Animation (t as color)');

% Initial scatter plot (t=0)
initial_vals = four_var_func(x_flat, y_flat, z_flat, 0);
scatter_obj = scatter3(x_flat, y_flat, z_flat, 10, initial_vals, 'filled');
colorbar('Label', 'Function Value');
colormap(viridis);

% Animation parameters
num_frames = 100;
time_step = 0.1;
frames = cell(num_frames, 1);

% Generate and play animation
for frame = 1:num_frames
    t = (frame - 1) * time_step;
    current_vals = four_var_func(x_flat, y_flat, z_flat, t);
    scatter_obj.CData = current_vals;
    title(['4-Variable Function Animation (t = ', num2str(t, '%.1f'), ')']);
    frames{frame} = getframe(gcf);
    drawnow; % Update plot in real-time
end

movie(frames, 1, 20); % Play once at 20 FPS

% Optional: Save animation to file
% writer = VideoWriter('4var_animation.avi');
% writer.FrameRate = 20;
% open(writer);
% writeVideo(writer, frames);
% close(writer);

Key Notes #

• Adjust grid resolution (20 in linspace) based on your system's performance.
• Use colormap to switch color scales (e.g., plasma, coolwarm).
• Modify time_step and num_frames to control animation duration and speed.

• Precompute Values: If your function is computationally expensive, precompute all function values for every time step upfront to speed up playback.
• Random Points: Instead of a full grid, use random 3D points to reduce computational load while maintaining visual clarity.
• Consistent Color Scale: Normalize function values (e.g., to [0,1] or [-1,1]) if their range varies drastically over time, ensuring the color scale stays consistent across frames.

内容的提问来源于stack exchange，提问作者NotaChoice