tomodrgn.analysis.get_ind_for_cluster

get_ind_for_cluster(labels: ndarray, selected_clusters: list[int]) → ndarray

Get the indices of particles belonging to the selected clusters.

Parameters:

• labels – array of cluster labels for each particle, shape (nptcls)

• selected_clusters – list of selected cluster labels

Returns:

array of indices of particles belonging to the selected clusters