Add node id to plot

Author: edeno

src/track_linearization/utils.py

@@ -60,6 +60,15 @@ def plot_track_graph(track_graph, ax=None, draw_edge_labels=False, **kwds):
         ax = plt.gca()
     node_position = nx.get_node_attributes(track_graph, "pos")
     nx.draw_networkx_nodes(track_graph, node_position, ax=ax, **kwds)
+    for node_id, pos in node_position.items():
+        plt.text(
+            pos[0],
+            pos[1],
+            str(node_id),
+            fontsize=8,
+            ha="center",
+            va="center",
+        )
     for node_id1, node_id2 in track_graph.edges:
         pos = np.stack((node_position[node_id1], node_position[node_id2]))
         ax.plot(pos[:, 0], pos[:, 1], color="black", zorder=-1)