James McCool commited on
Commit
31ba272
·
1 Parent(s): e560f1d

Fix column renaming logic in load_contest_file.py for MLB data processing

Browse files

- Updated the column renaming logic to use integer keys instead of string keys, ensuring accurate mapping of player positions for MLB contests.
- Enhanced data cleaning by maintaining consistency in column naming conventions, improving clarity in the dataset.

Files changed (1) hide show
  1. global_func/load_contest_file.py +1 -1
global_func/load_contest_file.py CHANGED
@@ -47,7 +47,7 @@ def load_contest_file(upload, sport):
47
  df[i] = df[i].str.replace(r'\s+(' + '|'.join(pos_values) + r')$', '', regex=True)
48
 
49
  if sport == 'MLB':
50
- df = df.rename(columns={'1': '1B', '2': '2B', '3': '3B', '4': 'C', '5': 'OF1', '6': 'OF2', '7': 'OF3', '8': 'SP1', '9': 'SP2', '10': 'SS'})
51
  ownership_dict = dict(zip(df['Player'], df['Own']))
52
  cleaned_df = df.drop(columns=['EntryId', 'EntryName', 'TimeRemaining', 'Points', 'Lineup', 'Player', 'Pos', 'Own', 'FPTS'])
53
  cleaned_df = cleaned_df[['BaseName', 'EntryCount', 'SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3']]
 
47
  df[i] = df[i].str.replace(r'\s+(' + '|'.join(pos_values) + r')$', '', regex=True)
48
 
49
  if sport == 'MLB':
50
+ df = df.rename(columns={1: '1B', 2: '2B', 3: '3B', 4: 'C', 5: 'OF1', 6: 'OF2', 7: 'OF3', 8: 'SP1', 9: 'SP2', 10: 'SS'})
51
  ownership_dict = dict(zip(df['Player'], df['Own']))
52
  cleaned_df = df.drop(columns=['EntryId', 'EntryName', 'TimeRemaining', 'Points', 'Lineup', 'Player', 'Pos', 'Own', 'FPTS'])
53
  cleaned_df = cleaned_df[['BaseName', 'EntryCount', 'SP1', 'SP2', 'C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3']]