Spaces:
Sleeping
Sleeping
James McCool
commited on
Commit
·
fb3a403
1
Parent(s):
8976120
Enhance data handling in load_ss_file.py by replacing zeros with NaN and dropping rows with any NaN values. This improves data integrity for CSV and Excel file imports.
Browse files
global_func/load_ss_file.py
CHANGED
|
@@ -21,12 +21,14 @@ def load_ss_file(lineups, csv_file):
|
|
| 21 |
|
| 22 |
if clean_name.endswith('.csv'):
|
| 23 |
lineups_df = pd.read_csv(lineups)
|
|
|
|
| 24 |
elif clean_name.endswith(('.xls', '.xlsx')):
|
| 25 |
lineups_df = pd.read_excel(lineups)
|
|
|
|
| 26 |
else:
|
| 27 |
st.error('Please upload either a CSV or Excel file for lineups')
|
| 28 |
return None, None
|
| 29 |
-
|
| 30 |
export_df = lineups_df.copy()
|
| 31 |
|
| 32 |
# Map the IDs to names
|
|
|
|
| 21 |
|
| 22 |
if clean_name.endswith('.csv'):
|
| 23 |
lineups_df = pd.read_csv(lineups)
|
| 24 |
+
lineups_df = lineups_df.replace(0, np.nan)
|
| 25 |
elif clean_name.endswith(('.xls', '.xlsx')):
|
| 26 |
lineups_df = pd.read_excel(lineups)
|
| 27 |
+
lineups_df = lineups_df.replace(0, np.nan)
|
| 28 |
else:
|
| 29 |
st.error('Please upload either a CSV or Excel file for lineups')
|
| 30 |
return None, None
|
| 31 |
+
lineups_df = lineups_df.dropna(how='any')
|
| 32 |
export_df = lineups_df.copy()
|
| 33 |
|
| 34 |
# Map the IDs to names
|