Refactor trim_portfolio function in global_func/trim_portfolio.py: optimize iteration logic for filtering portfolios based on ownership type.

global_func/trim_portfolio.py

@@ -2,11 +2,11 @@ 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)
+    for port_rows in range(len(working_portfolio)):
+        if working_portfolio.iloc[port_rows][own_type] > curr_own_type_max:
+            working_portfolio = working_portfolio.drop(port_rows)
             curr_own_type_max = working_portfolio[own_type].max()
-        elif row[performance_type] >= curr_own_type_max:
+        elif working_portfolio.iloc[port_rows][own_type] <= curr_own_type_max:
             pass
 
     return working_portfolio