function compute_index_states

deeptime.markov.sample.compute_index_states(dtrajs, subset=None) → List[ndarray]

Generates trajectory/time indices for the given list of states

Parameters:

• dtrajs (array_like or list of array_like) – Discretized trajectory or list of discretized trajectories. Negative elements will be ignored

• subset (ndarray((n)), optional, default = None) – array of states to be indexed. By default all states in dtrajs will be used

Returns:

indices – For each state, all trajectory and time indices where this state occurs. Each matrix has a number of rows equal to the number of occurrences of the corresponding state, with rows consisting of a tuple (i, t), where i is the index of the trajectory and t is the time index within the trajectory.

Return type:

List[np.ndarray]