James McCool commited on
Commit
17a3b09
·
1 Parent(s): 3b056d9

Refactor position assignment logic in `load_file.py` to improve data extraction

Browse files

- Updated the position assignment loop to use index-based iteration instead of direct value iteration, enhancing clarity and ensuring accurate mapping of player positions from the `Lineup` column.

Files changed (1) hide show
  1. global_func/load_file.py +2 -2
global_func/load_file.py CHANGED
@@ -22,8 +22,8 @@ def load_file(upload):
22
  df = raw_df[['EntryId', 'EntryName', 'TimeRemaining', 'Points', 'Lineup', 'Player', 'Roster Position', '%Drafted', 'FPTS']]
23
  df = df.rename(columns={'Roster Position': 'Pos', '%Drafted': 'Own'})
24
  df['Lineup'] = df['Lineup'].replace(pos_values, [',']*len(pos_values), regex=True)
25
- for pos in pos_values:
26
- df[pos] = df['Lineup'].str.split(',').str[pos_values.index(pos)]
27
  position_dict = dict(zip(df['Player'], df['Pos']))
28
  ownership_dict = dict(zip(df['Player'], df['Own']))
29
  entry_list = list(set(df['EntryName']))
 
22
  df = raw_df[['EntryId', 'EntryName', 'TimeRemaining', 'Points', 'Lineup', 'Player', 'Roster Position', '%Drafted', 'FPTS']]
23
  df = df.rename(columns={'Roster Position': 'Pos', '%Drafted': 'Own'})
24
  df['Lineup'] = df['Lineup'].replace(pos_values, [',']*len(pos_values), regex=True)
25
+ for i in range(0,9):
26
+ df[pos_values[i]] = df['Lineup'].str.split(',').str[i]
27
  position_dict = dict(zip(df['Player'], df['Pos']))
28
  ownership_dict = dict(zip(df['Player'], df['Own']))
29
  entry_list = list(set(df['EntryName']))