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.
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={
|
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']]
|