diff options
Diffstat (limited to 'ero1/src/helper/duplicate_removal.py')
| -rw-r--r-- | ero1/src/helper/duplicate_removal.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/ero1/src/helper/duplicate_removal.py b/ero1/src/helper/duplicate_removal.py new file mode 100644 index 0000000..4138bb8 --- /dev/null +++ b/ero1/src/helper/duplicate_removal.py @@ -0,0 +1,16 @@ +from src.helper.debug_printer import debug_print + +def remove_duplicates(graph, debug_mode=False): + """ + Supprime les noeuds dupliqués du graphe. + Parameters: + graph : Le graphe à nettoyer. + debug_mode : Indique si le mode debug est activé ou non. + """ + + removed = [(u, v, k) for u, v, k in graph.edges(keys=True) if k > 0] + graph.remove_edges_from(removed) + + debug_print(f"Nombre d'arêtes dupliquées supprimées : {len(removed)}", debug_mode) + + return graph |
