tropea_clustering.plot_smooth.plot_one_trj_multi

tropea_clustering.plot_smooth.plot_one_trj_multi(title, example_id, input_data, labels)[source]

Plots the colored trajectory of an example particle.

Parameters:

  • title (pathlib.Path) – The path of the .png file the figure will be saved as.

  • example_id (int) – The ID of the selected particle.

  • input_data (ndarray of shape (n_particles, n_frames, n_features)) – The input data array.

  • labels (ndarray of shape (n_particles, n_frames)) – The output of the clustering algorithm.

Example

Here’s an example of the output:

Example Image

The datapoints are colored according to the cluster they have been assigned to.