James McCool commited on
Commit
d223b7a
·
1 Parent(s): 07e0b07

Refactor general exposures dataframe creation in create_general_exposures function

Browse files

- Updated the creation of the general exposures dataframe to use a more concise dictionary format, improving code readability.
- Changed the output method from st.table to st.write for better integration with Streamlit's display capabilities, enhancing user experience.

global_func/create_general_exposures.py CHANGED
@@ -25,10 +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
  st.write(f'{each_col} Sum is {each_general}, Average is {each_general / each_general_len_set[general_count_var]}')
28
- general_frame = pd.DataFrame()
29
- general_frame['Stat'] = each_col
30
- general_frame['Average'] = each_general / each_general_len_set[general_count_var]
31
- general_frame = general_frame[['Stat', 'Average']]
32
  st.write(general_frame)
33
  general_frame = general_frame.rename(columns={'Average': f'Average {each_set_name[general_count_var]}'})
34
  if len(general_exposures) == 0:
@@ -36,6 +36,6 @@ def create_general_exposures(df: pd.DataFrame, entrants: list = None):
36
  else:
37
  general_exposures = pd.merge(general_exposures, general_frame, on='Stat', how='outer')
38
  general_count_var += 1
39
- st.table(general_exposures)
40
 
41
  return general_exposures
 
25
  general_count_var = 0
26
  for each_general in each_general_set:
27
  st.write(f'{each_col} Sum is {each_general}, Average is {each_general / each_general_len_set[general_count_var]}')
28
+ general_frame = pd.DataFrame({
29
+ 'Stat': [each_col],
30
+ 'Average': [each_general / each_general_len_set[general_count_var]]
31
+ })
32
  st.write(general_frame)
33
  general_frame = general_frame.rename(columns={'Average': f'Average {each_set_name[general_count_var]}'})
34
  if len(general_exposures) == 0:
 
36
  else:
37
  general_exposures = pd.merge(general_exposures, general_frame, on='Stat', how='outer')
38
  general_count_var += 1
39
+ st.write(general_exposures)
40
 
41
  return general_exposures