From bd9db227011eefc9d691ed6b168886cbc00362a7 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 22 Oct 2024 14:01:22 +0200
Subject: [PATCH] datasets

---
 prosit_data_merge.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/prosit_data_merge.py b/prosit_data_merge.py
index 1211691..d864658 100644
--- a/prosit_data_merge.py
+++ b/prosit_data_merge.py
@@ -150,7 +150,8 @@ df = pd.read_pickle('database/data_prosit_merged_holdout.pkl')
 
 print(len(df))
 print(df.head())
-df = df[df['Retention time']!=[0,0,0,0,0,0]]
+df['Retention time']=df['Retention time'].apply(lambda x : x[0])
+df = df.loc[df['Retention time']!=0]
 print(len(df))
 print(df.head())
 
-- 
GitLab