James McCool commited on
Commit
e7bbe09
·
1 Parent(s): 9ca8f43

Refactor map_mask_parse calls to use player names directly instead of mapped IDs for DraftKings and FanDuel, improving clarity and consistency in data processing.

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +13 -12
src/streamlit_app.py CHANGED
@@ -518,6 +518,7 @@ def normalize_special_characters(text):
518
 
519
  @st.cache_data
520
  def map_mask_parse(df: pd.DataFrame, map: dict, threshold: float, site_var: str):
 
521
  if site_var == 'Draftkings':
522
  proj_df = df.iloc[:, :-7].replace(map).astype(float)
523
  elif site_var == 'Fanduel':
@@ -1275,22 +1276,22 @@ if selected_tab == 'Optimals':
1275
  data_export = data_export[data_export['Team_count'] >= min_stacks_var]
1276
  data_export = data_export[data_export['Team_count'] <= max_stacks_var]
1277
  if site_var == 'Draftkings':
1278
- data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Median'])), projection_var, site_var)
1279
- data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Own'])), ownership_var, site_var)
1280
  elif site_var == 'Fanduel':
1281
- data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Median'])), projection_var, site_var)
1282
- data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Own'])), ownership_var, site_var)
1283
 
1284
  name_export = name_export[name_export['salary'] >= salary_min_var]
1285
  name_export = name_export[name_export['salary'] <= salary_max_var]
1286
  name_export = name_export[name_export['Team_count'] >= min_stacks_var]
1287
  name_export = name_export[name_export['Team_count'] <= max_stacks_var]
1288
  if site_var == 'Draftkings':
1289
- name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Median'])), projection_var, site_var)
1290
- name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Own'])), ownership_var, site_var)
1291
  elif site_var == 'Fanduel':
1292
- name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Median'])), projection_var, site_var)
1293
- name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Own'])), ownership_var, site_var)
1294
 
1295
  reg_opt_col, pm_opt_col = st.columns(2)
1296
  with reg_opt_col:
@@ -1367,11 +1368,11 @@ if selected_tab == 'Optimals':
1367
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['salary'] >= salary_min_var]
1368
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['salary'] <= salary_max_var]
1369
  if site_var == 'Draftkings':
1370
- st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Median'])), projection_var, site_var)
1371
- st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(dk_id_map), raw_baselines['Own'])), ownership_var, site_var)
1372
  elif site_var == 'Fanduel':
1373
- st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Median'])), projection_var, site_var)
1374
- st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'].map(fd_id_map), raw_baselines['Own'])), ownership_var, site_var)
1375
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['Team_count'] >= min_stacks_var]
1376
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['Team_count'] <= max_stacks_var]
1377
  st.session_state.data_export_display = st.session_state.data_export_display.reset_index(drop=True)
 
518
 
519
  @st.cache_data
520
  def map_mask_parse(df: pd.DataFrame, map: dict, threshold: float, site_var: str):
521
+ print(proj_df.head(10))
522
  if site_var == 'Draftkings':
523
  proj_df = df.iloc[:, :-7].replace(map).astype(float)
524
  elif site_var == 'Fanduel':
 
1276
  data_export = data_export[data_export['Team_count'] >= min_stacks_var]
1277
  data_export = data_export[data_export['Team_count'] <= max_stacks_var]
1278
  if site_var == 'Draftkings':
1279
+ data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var, site_var)
1280
+ data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var, site_var)
1281
  elif site_var == 'Fanduel':
1282
+ data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var, site_var)
1283
+ data_export = map_mask_parse(data_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var, site_var)
1284
 
1285
  name_export = name_export[name_export['salary'] >= salary_min_var]
1286
  name_export = name_export[name_export['salary'] <= salary_max_var]
1287
  name_export = name_export[name_export['Team_count'] >= min_stacks_var]
1288
  name_export = name_export[name_export['Team_count'] <= max_stacks_var]
1289
  if site_var == 'Draftkings':
1290
+ name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var, site_var)
1291
+ name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var, site_var)
1292
  elif site_var == 'Fanduel':
1293
+ name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var, site_var)
1294
+ name_export = map_mask_parse(name_export, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var, site_var)
1295
 
1296
  reg_opt_col, pm_opt_col = st.columns(2)
1297
  with reg_opt_col:
 
1368
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['salary'] >= salary_min_var]
1369
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['salary'] <= salary_max_var]
1370
  if site_var == 'Draftkings':
1371
+ st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var, site_var)
1372
+ st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var, site_var)
1373
  elif site_var == 'Fanduel':
1374
+ st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Median'])), projection_var, site_var)
1375
+ st.session_state.data_export_display = map_mask_parse(st.session_state.data_export_display, dict(zip(raw_baselines['Player'], raw_baselines['Own'])), ownership_var, site_var)
1376
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['Team_count'] >= min_stacks_var]
1377
  st.session_state.data_export_display = st.session_state.data_export_display[st.session_state.data_export_display['Team_count'] <= max_stacks_var]
1378
  st.session_state.data_export_display = st.session_state.data_export_display.reset_index(drop=True)