James McCool commited on
Commit
f69b920
·
1 Parent(s): 8ee5fa8

Refactor data export logic to utilize list comprehension for mapping player IDs, improving performance and readability

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -249,8 +249,8 @@ with tab2:
249
  st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
250
  st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
251
  data_export = st.session_state.working_seed.copy()
252
- for col in data_export[:, 0:9]:
253
- data_export[col] = data_export[col].map(dk_id_dict)
254
  st.download_button(
255
  label="Export optimals set",
256
  data=convert_df(data_export),
 
249
  st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
250
  st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
251
  data_export = st.session_state.working_seed.copy()
252
+ for col in range(9):
253
+ data_export[:, col] = np.array([dk_id_dict.get(x, x) for x in data_export[:, col]])
254
  st.download_button(
255
  label="Export optimals set",
256
  data=convert_df(data_export),