replay_trajectory_classification.environments.extract_bin_info_from_track_graph#

extract_bin_info_from_track_graph(track_graph: Graph, track_graph_with_bin_centers_edges: Graph, edge_order: list[tuple], edge_spacing: float | list) → DataFrame[source]#

For each node, find edge_id, is_bin_edge, x_position, y_position, and linear_position.

Parameters:

  • track_graph (nx.Graph)

  • track_graph_with_bin_centers_edges (nx.Graph)

  • edge_order (list of 2-tuples)

  • edge_spacing (list, len n_edges - 1)

Returns:

nodes_df – Collect information about each bin

Return type:

pd.DataFrame