dynamo.pl.plot_traj

dynamo.pl.plot_traj(f, y0, t, args=(), lw=2, background=None, integration_direction='both', save_show_or_return='return', save_kwargs={}, ax=None)[source]

Plots a trajectory on a phase portrait. code adapted from: http://be150.caltech.edu/2017/handouts/dynamical_systems_approaches.html

Parameters:

  • f (function for form f(y, t, *args)) – The right-hand-side of the dynamical system. Must return a 2-array.

  • y0 (array_like, shape (2,)) – Initial condition.

  • t (array_like) – Time points for trajectory.

  • args (tuple, default ()) – Additional arguments to be passed to f

  • lw (float, (default: 2)) – The line width of the trajectory.

  • background (str or None (default: None)) – The background color of the plot.

  • integration_direction (str (default: forward)) – Integrate the trajectory in forward, backward or both directions. default is ‘both’.

  • save_show_or_return ({‘show’, ‘save’, ‘return’} (default: return)) – Whether to save, show or return the figure.

  • save_kwargs (dict (default: {})) – A dictionary that will passed to the save_fig function. By default it is an empty dictionary and the save_fig function will use the {“path”: None, “prefix”: ‘plot_traj’, “dpi”: None, “ext”: ‘pdf’, “transparent”: True, “close”: True, “verbose”: True} as its parameters. Otherwise you can provide a dictionary that properly modify those keys according to your needs.

  • ax (Matplotlib Axis instance) – Axis on which to make the plot

Returns:

output – Axis with streamplot included.

Return type:

Matplotlib Axis instance