Update data_processor.py
Browse files- data_processor.py +16 -2
data_processor.py
CHANGED
@@ -182,6 +182,15 @@
|
|
182 |
|
183 |
|
184 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
185 |
import re
|
186 |
import pandas as pd
|
187 |
import os
|
@@ -262,6 +271,11 @@ class DataProcessor:
|
|
262 |
return column
|
263 |
raise ValueError("No intervention column found in the dataframe.")
|
264 |
|
|
|
|
|
|
|
|
|
|
|
265 |
def compute_intervention_statistics(self, df):
|
266 |
intervention_column = self.find_intervention_column(df)
|
267 |
total_days = len(df)
|
@@ -284,9 +298,9 @@ class DataProcessor:
|
|
284 |
return self.NOT_ENGAGED_STR
|
285 |
else:
|
286 |
return 'Unknown'
|
287 |
-
|
288 |
def compute_student_metrics(self, df):
|
289 |
-
intervention_column = self.
|
290 |
intervention_df = df[df[intervention_column].str.strip().str.lower() == 'yes']
|
291 |
intervention_sessions_held = len(intervention_df)
|
292 |
student_columns = [col for col in df.columns if col.startswith('Student Attendance')]
|
|
|
182 |
|
183 |
|
184 |
|
185 |
+
|
186 |
+
|
187 |
+
|
188 |
+
|
189 |
+
|
190 |
+
|
191 |
+
|
192 |
+
|
193 |
+
|
194 |
import re
|
195 |
import pandas as pd
|
196 |
import os
|
|
|
271 |
return column
|
272 |
raise ValueError("No intervention column found in the dataframe.")
|
273 |
|
274 |
+
def get_intervention_column(self, df):
|
275 |
+
if self.intervention_column is None:
|
276 |
+
self.intervention_column = self.find_intervention_column(df)
|
277 |
+
return self.intervention_column
|
278 |
+
|
279 |
def compute_intervention_statistics(self, df):
|
280 |
intervention_column = self.find_intervention_column(df)
|
281 |
total_days = len(df)
|
|
|
298 |
return self.NOT_ENGAGED_STR
|
299 |
else:
|
300 |
return 'Unknown'
|
301 |
+
|
302 |
def compute_student_metrics(self, df):
|
303 |
+
intervention_column = self.get_intervention_column(df)
|
304 |
intervention_df = df[df[intervention_column].str.strip().str.lower() == 'yes']
|
305 |
intervention_sessions_held = len(intervention_df)
|
306 |
student_columns = [col for col in df.columns if col.startswith('Student Attendance')]
|