msmtools.dtraj.sample_indexes_by_state

msmtools.dtraj.sample_indexes_by_state(indexes, nsample, subset=None, replace=True)

Samples trajectory/time indexes according to the given sequence of states

Parameters:

• indexes (list of ndarray( (N_i, 2) )) – For each state, all trajectory and time indexes 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.
• nsample (int) – Number of samples per state. If replace = False, the number of returned samples per state could be smaller if less than nsample indexes are available for a state.
• subset (ndarray((n)), optional, default = None) – array of states to be indexed. By default all states in dtrajs will be used
• replace (boolean, optional) – Whether the sample is with or without replacement

Returns:

indexes – List of the sampled indices by state. Each element is an index array with a number of rows equal to N=len(sequence), 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, 2) )