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')]
|