diff --git a/script/base/dataframe_base.py b/script/base/dataframe_base.py index 9fd9b472fc1ad340bdfb09cd55eeb44d26221656..6a70c5a2f2734bcee984656be45539372e2d54a7 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):