Add 'Trim Portfolio' button in app.py and implement trim_portfolio function in new global_func/trim_portfolio.py

app.py

@@ -15,6 +15,7 @@ from global_func.predict_dupes import predict_dupes
 from global_func.highlight_rows import highlight_changes, highlight_changes_winners, highlight_changes_losers
 from global_func.load_csv import load_csv
 from global_func.find_csv_mismatches import find_csv_mismatches
+from global_func.trim_portfolio import trim_portfolio
 
 freq_format = {'Finish_percentile': '{:.2%}', 'Lineup Edge': '{:.2%}', 'Win%': '{:.2%}'}
 player_wrong_names_mlb = ['Enrique Hernandez']
@@ -779,6 +780,8 @@ with tab3:
             )
 
         col1, col2 = st.columns([1, 10])
+        if st.button('Trim Portfolio'):
+            st.session_state['portfolio'] = trim_portfolio(st.session_state['portfolio'], 'median', 'Own')
         with col1:
             with st.form(key='filter_form'):
                 max_dupes = st.number_input("Max acceptable dupes?", value=1000, min_value=1, step=1)

global_func/trim_portfolio.py

@@ -0,0 +1,12 @@
+def trim_portfolio(portfolio, performance_type, own_type):
+    working_portfolio = portfolio.sort_values(by=performance_type, ascending = False)
+    curr_own_type_max = working_portfolio[own_type].max()
+
+    for row in working_portfolio.iterrows():
+        if row[performance_type] < curr_own_type_max:
+            working_portfolio = working_portfolio.drop(row.name)
+            curr_own_type_max = working_portfolio[own_type].max()
+        elif row[performance_type] >= curr_own_type_max:
+            pass
+
+    return working_portfolio