Attach label to node

GEMDAT-repos · Jun 4, 2024 · d11abfa · d11abfa
1 parent af3b095
commit d11abfa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/gemdat/jumps.py b/src/gemdat/jumps.py
@@ -375,6 +375,9 @@ def to_graph(
 
         G = nx.DiGraph()
 
+        for i, site in enumerate(self.sites):
+            G.add_node(i, label=site.label)
+
         for (start, stop), n_jumps in self._counter().items():
             time_perc = atom_percentage[start] * self.trajectory.total_time