slitflow.trj.filter module

class StepAtLeast(info_path=None)

Bases: Table

Select trajectories by the step number.

Parameters:

• reqs[0] (Table) – Trajectory data.

• param["step"] (int) – Step number for selection. If you set step=2, trajectories containing at least three points (frames) are selected.

• param["group_depth"] (int) – Depth number of trajectory number column.

• param["split_depth"] (int) – File split depth number.

Returns:

Selected trajectory Table

Return type:

Table

set_info(param={})

Copy info from reqs[0] and add params.

static process(reqs, param)

Select trajectories by the step number.

Parameters:

• reqs[0] (pandas.DataFrame) – Trajectory data.

• param["index_cols"] (list of str) – Index column names for pandas.DataFrame.groupby(). List should include the trajectory number column.

Returns:

Selected trajectory table

Return type:

pandas.DataFrame

class StepRange(info_path=None)

Bases: Table

Select trajectories by the step number range.

Parameters:

• reqs[0] (Table) – Trajectory data.

• param["step_range"] (list of int) – Minimum and maximum numbers of trajectory step.

• param["group_depth"] (int) – Data split depth number.

• param["split_depth"] (int) – File split depth number.

Returns:

Selected trajectory Table

Return type:

Table

set_info(param={})

Copy info from reqs[0] and add params.

static process(reqs, param)

Select trajectories by the step number range.

Parameters:

• reqs[0] (pandas.DataFrame) – Trajectory data.

• param["step_range"] (list of int) – Minimum and maximum numbers of trajectory step.

• param["index_cols"] (list of str) – Index column names for pandas.DataFrame.groupby(). List should include the trajectory number column.

Returns:

Selected trajectory table

Return type:

pandas.DataFrame