ProfessorLeVesseur commited on
Commit
a280495
·
verified ·
1 Parent(s): 52646f5

Update data_processor.py

Browse files
Files changed (1) hide show
  1. 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.find_intervention_column(df)
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')]