msmtools.dtraj.index_states

msmtools.dtraj.index_states(dtrajs, subset=None)

Generates a trajectory/time indexes for the given list of states

Parameters:

• dtraj (array_like or list of array_like) – Discretized trajectory or list of discretized trajectories
• subset (ndarray((n)), optional, default = None) – array of states to be indexed. By default all states in dtrajs will be used

Returns:

indexes – For each state, all trajectory and time indexes where this state occurs. Each matrix has a number of rows equal to the number of occurances 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 of ndarray( (N_i, 2) )