From d2e6dcc572a00d49c475a9957486e5f31018f65b Mon Sep 17 00:00:00 2001
From: Fontrodona Nicolas <nicolas.fontrodona@ens-lyon.fr>
Date: Tue, 12 Jan 2021 14:59:40 +0100
Subject: [PATCH]
src/find_interaction_cluster/community_figures/fig_functions.py: fix
get_feature_by_community function
---
src/find_interaction_cluster/community_figures/fig_functions.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/find_interaction_cluster/community_figures/fig_functions.py b/src/find_interaction_cluster/community_figures/fig_functions.py
index 20747cae..e2a09988 100644
--- a/src/find_interaction_cluster/community_figures/fig_functions.py
+++ b/src/find_interaction_cluster/community_figures/fig_functions.py
@@ -466,6 +466,8 @@ def get_feature_by_community(df: pd.DataFrame, feature: str) -> Dict:
dic = {}
for i in range(df.shape[0]):
com, id_ft = df.iloc[i, :][['community', f'id_{feature}']]
+ if isinstance(com, float):
+ com = None if np.isnan(com) else com
if com is not None:
if com in dic:
dic[com].append(id_ft)
--
GitLab