James McCool commited on
Commit
42199ca
·
1 Parent(s): 22483d5

Limit concatenation in distribute_preset function to the top 10 entries of the working portfolio, removing debug print statements for cleaner output and improved performance in portfolio processing.

Browse files
Files changed (1) hide show
  1. global_func/distribute_preset.py +1 -4
global_func/distribute_preset.py CHANGED
@@ -19,9 +19,6 @@ def distribute_preset(portfolio: pd.DataFrame, lineup_target: int, exclude_cols:
19
  working_portfolio = working_portfolio[working_portfolio['Finish_percentile'] >= lower_threshold]
20
  working_portfolio = working_portfolio.sort_values(by='median', ascending = False)
21
  working_portfolio = working_portfolio.reset_index(drop=True)
22
- print(finish_threshold)
23
- print(lower_threshold)
24
- print(working_portfolio[['Finish_percentile', 'Weighted Own']].sort_values(by='Finish_percentile', ascending=True).head(10))
25
  if len(working_portfolio) == 0:
26
  continue
27
  elif len(working_portfolio) >= 1:
@@ -34,7 +31,7 @@ def distribute_preset(portfolio: pd.DataFrame, lineup_target: int, exclude_cols:
34
  curr_own_type_max = working_portfolio.loc[i, 'Weighted Own'] + (slack_var / 20 * working_portfolio.loc[i, 'Weighted Own'])
35
 
36
  working_portfolio = working_portfolio.drop(rows_to_drop).reset_index(drop=True)
37
- concat_portfolio = pd.concat([concat_portfolio, working_portfolio])
38
 
39
  if len(concat_portfolio) >= lineup_target:
40
  return concat_portfolio.sort_values(by='Finish_percentile', ascending=True).head(lineup_target)
 
19
  working_portfolio = working_portfolio[working_portfolio['Finish_percentile'] >= lower_threshold]
20
  working_portfolio = working_portfolio.sort_values(by='median', ascending = False)
21
  working_portfolio = working_portfolio.reset_index(drop=True)
 
 
 
22
  if len(working_portfolio) == 0:
23
  continue
24
  elif len(working_portfolio) >= 1:
 
31
  curr_own_type_max = working_portfolio.loc[i, 'Weighted Own'] + (slack_var / 20 * working_portfolio.loc[i, 'Weighted Own'])
32
 
33
  working_portfolio = working_portfolio.drop(rows_to_drop).reset_index(drop=True)
34
+ concat_portfolio = pd.concat([concat_portfolio, working_portfolio.head(10)])
35
 
36
  if len(concat_portfolio) >= lineup_target:
37
  return concat_portfolio.sort_values(by='Finish_percentile', ascending=True).head(lineup_target)