diff --git a/mesures/f3.py b/mesures/f3.py index 98f75d6..16d3543 100644 --- a/mesures/f3.py +++ b/mesures/f3.py @@ -99,9 +99,9 @@ def reader(self, filepath): 'ae': 'sum' } ).reset_index() - # TODO review obtencion and firmeza - df['method'] = 1 - df['firmeza'] = 1 + df['firmeza'] = df.apply(lambda row: 1 if row['method'] in (1, 3) else 0, axis=1) + # TODO Review method + df = df[columns] return df