diff --git a/parse.py b/parse.py index 525905c..c04b6ef 100755 --- a/parse.py +++ b/parse.py @@ -45,16 +45,17 @@ def parse_infond(): df = df.filter(['SKLAD', 'VEP']) # Cleanup the "SKLAD" name - a = df.at[23,'SKLAD'] - df.at[23,'SKLAD'] = a.split()[0]+' '+a.split()[1] - df.at[23,'VEP'] = a.split()[2] + a = df.at[11,'SKLAD'] + df.at[11,'SKLAD'] = a.split()[0]+' '+a.split()[1] + df.at[11,'VEP'] = a.split()[2] - a = df.at[15,'SKLAD'] - df.at[15,'SKLAD'] = a.split()[0]+' '+a.split()[1] - df.at[15,'VEP'] = a.split()[2] + a = df.at[19,'SKLAD'] + df.at[19,'SKLAD'] = a.split()[0]+' '+a.split()[1]+' '+a.split()[2] + df.at[19,'VEP'] = a.split()[3] # Drop all rows except the ones we want - return df[(df["SKLAD"]=="Infond Defensive") | (df["SKLAD"] =="Infond Technology") ] + #return df[(df["SKLAD"]=="Infond Defensive") | (df["SKLAD"] =="Infond Technology") ] + return df[(df["SKLAD"]=="Infond Globalni defenzivni") | (df["SKLAD"] =="Infond Tehnologija") ] ###############################################################################