From f515deabea77657c8e2838e08b8bcbfb5b6ccdf6 Mon Sep 17 00:00:00 2001
From: MAFTOUH Mohammed Amine <mohammed-amine.maftouh@imt-atlantique.net>
Date: Mon, 3 Mar 2025 13:15:19 +0000
Subject: [PATCH] Add new file
---
Visu/affichage_messages_par_cluster.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
create mode 100644 Visu/affichage_messages_par_cluster.py
diff --git a/Visu/affichage_messages_par_cluster.py b/Visu/affichage_messages_par_cluster.py
new file mode 100644
index 0000000..c5faf12
--- /dev/null
+++ b/Visu/affichage_messages_par_cluster.py
@@ -0,0 +1,17 @@
+def afficher_messages_par_cluster(df,labels):
+ """
+ Ajoute les labels de cluster à la table et retourne les messages par cluster.
+
+ :param labels: Liste des labels de cluster déjà générés
+ :param df: DataFrame contenant la colonne 'message'
+ :return: Dictionnaire {cluster: [messages]}
+ """
+ df['cluster'] = labels
+ messages_par_cluster = {}
+
+ # Remplir le dictionnaire avec les messages de chaque cluster
+ for cluster in sorted(df['cluster'].unique()):
+ messages = df[df['cluster'] == cluster]['message'].tolist()
+ messages_par_cluster[cluster] = messages
+
+ return messages_par_cluster
\ No newline at end of file
--
GitLab