trajectory_3d

Contents

trajectory_3d#

class braintools.visualize.trajectory_3d(trajectory, time_colors=True, start_marker='o', end_marker='s', cmap='viridis', ax=None, figsize=(12, 10), title=None, **kwargs)#

Visualize 3D trajectory of neural state evolution.

Parameters:
  • trajectory (ndarray) – 3D trajectory of shape (time_steps, 3).

  • time_colors (bool) – Whether to color trajectory by time.

  • start_marker (str) – Markers for start and end points.

  • end_marker (str) – Markers for start and end points.

  • cmap (str) – Colormap for time coloring.

  • ax (Axes3D | None) – 3D axes to plot on.

  • figsize (Tuple[float, float]) – Figure size if creating new figure.

  • title (str | None) – Plot title.

  • **kwargs – Additional arguments passed to plot.

Returns:

ax – The 3D axes object.

Return type:

Axes3D