DFS_Portfolio_Manager / global_func /load_dk_fd_file.py
James McCool
Remove unnecessary columns from lineups DataFrame in load_dk_fd_file.py: drop 'Entry ID', 'Contest Name', 'Contest ID', and 'Entry Fee' to streamline data processing and enhance analysis efficiency.
cda266d
raw
history blame
1.12 kB
import streamlit as st
import numpy as np
import pandas as pd
import time
from fuzzywuzzy import process
def load_dk_fd_file(lineups, csv_file):
df = csv_file.copy()
try:
name_dict = dict(zip(df['Name + ID'], df['Name']))
except:
name_dict = dict(zip(df['Id'], df['Nickname']))
# Now load and process the lineups file
try:
if lineups.name.endswith('.csv'):
lineups_df = pd.read_csv(lineups)
elif lineups.name.endswith(('.xls', '.xlsx')):
lineups_df = pd.read_excel(lineups)
else:
st.error('Please upload either a CSV or Excel file for lineups')
return None, None
lineups_df = lineups_df.drop(columns=['Entry ID', 'Contest Name', 'Contest ID', 'Entry Fee'])
export_df = lineups_df.copy()
# Map the IDs to names
for col in lineups_df.columns:
lineups_df[col] = lineups_df[col].map(name_dict)
return export_df, lineups_df
except Exception as e:
st.error(f'Error loading lineups file: {str(e)}')
return None, None