From d192ed7347a11107715d4999e96601fcfe29be92 Mon Sep 17 00:00:00 2001 From: Legton <legton0@gmail.com> Date: Wed, 1 Nov 2017 13:07:15 +0000 Subject: [PATCH] arruma outros do dataframe --- script/base/dataframe_base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/script/base/dataframe_base.py b/script/base/dataframe_base.py index 9fd9b47..6a70c5a 100644 --- a/script/base/dataframe_base.py +++ b/script/base/dataframe_base.py @@ -95,9 +95,12 @@ def fix_admission(df): def fix_evasion(df): + evasionForms = [x[1] for x in EvasionForm.EVASION_FORM] + df.loc[~df.FORMA_EVASAO.isin(evasionForms), 'FORMA_EVASAO'] = 100 for evasion in EvasionForm.EVASION_FORM: #~ df.loc[df.FORMA_EVASAO.str.contains(evasion[1]).fillna(1.0), 'FORMA_EVASAO'] = evasion[0] df.loc[df.FORMA_EVASAO == evasion[1], 'FORMA_EVASAO'] = evasion[0] + #~ if(evasion[0] == 100): #~ for x in df.FORMA_EVASAO.str.contains(evasion[1]).fillna(False): #~ if(x != 0.0): -- GitLab