tropea_clustering.plot_smooth.plot_output_multi

tropea_clustering.plot_smooth.plot_output_multi(title, input_data, state_list, labels)[source]

Plot a cumulative figure showing trajectories and identified states.

Parameters:

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

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

  • state_list (list[StateMulti]) – The list of the cluster states.

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

Example

Example Image

All the data are plotted, colored according to the cluster thay have been assigned to. The clusters are shown as black ellipses, whose orizontal and vertical axis length is given by the standard deviation of the Gaussians corresponding to the cluster. Unclassififed data points are colored in purple.