DFS_Portfolio_Manager / global_func /get_portfolio_names.py
James McCool
Refactor import statements across multiple files to replace 'fuzzywuzzy' with 'rapidfuzz' for improved performance and consistency in string matching functionality. Additionally, clean up unused imports in app.py and related global functions to enhance code clarity and maintainability.
d9db89f
raw
history blame
863 Bytes
import streamlit as st
import numpy as np
import pandas as pd
import time
from rapidfuzz import process
def get_portfolio_names(portfolio_df):
"""
Get all unique names from the portfolio dataframe's player columns.
Args:
portfolio_df: DataFrame containing portfolio data
Returns:
list: List of unique player names
"""
# Get columns that contain player names (excluding non-player columns)
player_columns = [col for col in portfolio_df.columns
if col not in ['salary', 'median', 'Own']]
# Get all unique values from these columns
unique_names = portfolio_df[player_columns].values.flatten()
unique_names = pd.unique(unique_names) # Remove duplicates
unique_names = unique_names[~pd.isna(unique_names)] # Remove any NaN values
return list(unique_names)