DFS_Portfolio_Manager / global_func /get_portfolio_names.py
James McCool
ping
9326be6
raw
history blame contribute delete
929 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
"""
player_columns = [col for col in portfolio_df.columns
if col not in ['salary', 'median', 'Own', 'Dupes', 'Stack', 'Size', 'Finish%', 'Finish_percentile', 'Win%', 'Edge', 'Lineup Edge', 'Weighted Own', 'Geomean', 'Diversity', 'SE Score']]
# 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)