James McCool commited on
Commit
00d6b89
·
1 Parent(s): 21f63dc

Handle division by zero in create_general_exposures function

Browse files

- Added a try-except block to manage potential division by zero errors when calculating averages, ensuring that the function assigns a value of 0 in such cases.
- Maintained existing functionality while improving the robustness of the data processing logic.

global_func/create_general_exposures.py CHANGED
@@ -25,7 +25,10 @@ def create_general_exposures(df: pd.DataFrame, entrants: list = None):
25
  general_count_var = 0
26
  for each_general in each_general_set:
27
  general_frame['Stat'] = [each_col]
28
- general_frame['Average'] = [each_general / each_general_len_set[general_count_var]]
 
 
 
29
  general_frame = general_frame.rename(columns={'Average': f'Average {each_set_name[general_count_var]}'})
30
  if len(general_exposures) == 0:
31
  general_row = general_frame
 
25
  general_count_var = 0
26
  for each_general in each_general_set:
27
  general_frame['Stat'] = [each_col]
28
+ try:
29
+ general_frame['Average'] = [each_general / each_general_len_set[general_count_var]]
30
+ except:
31
+ general_frame['Average'] = [0]
32
  general_frame = general_frame.rename(columns={'Average': f'Average {each_set_name[general_count_var]}'})
33
  if len(general_exposures) == 0:
34
  general_row = general_frame