Update functions/df_update.py
Browse files- functions/df_update.py +1 -30
functions/df_update.py
CHANGED
|
@@ -229,36 +229,7 @@ class df_update:
|
|
| 229 |
# Check if 'trajectory_null' column exists and drop it
|
| 230 |
if 'trajectory_null' in df.columns:
|
| 231 |
df = df.drop('trajectory_null')
|
| 232 |
-
|
| 233 |
-
|
| 234 |
-
pitch_cat = {'FA': None,
|
| 235 |
-
'FF': 'Fastball',
|
| 236 |
-
'FT': 'Fastball',
|
| 237 |
-
'FC': 'Fastball',
|
| 238 |
-
'FS': 'Off-Speed',
|
| 239 |
-
'FO': 'Off-Speed',
|
| 240 |
-
'SI': 'Fastball',
|
| 241 |
-
'ST': 'Breaking',
|
| 242 |
-
'SL': 'Breaking',
|
| 243 |
-
'CU': 'Breaking',
|
| 244 |
-
'KC': 'Breaking',
|
| 245 |
-
'SC': 'Off-Speed',
|
| 246 |
-
'GY': 'Off-Speed',
|
| 247 |
-
'SV': 'Breaking',
|
| 248 |
-
'CS': 'Breaking',
|
| 249 |
-
'CH': 'Off-Speed',
|
| 250 |
-
'KN': 'Off-Speed',
|
| 251 |
-
'EP': 'Breaking',
|
| 252 |
-
'UN': None,
|
| 253 |
-
'IN': None,
|
| 254 |
-
'PO': None,
|
| 255 |
-
'AB': None,
|
| 256 |
-
'AS': None,
|
| 257 |
-
'NP': None}
|
| 258 |
-
df = df.with_columns(
|
| 259 |
-
df["pitch_type"].map_elements(lambda x: pitch_cat.get(x, x)).alias("pitch_group")
|
| 260 |
-
)
|
| 261 |
-
|
| 262 |
return df
|
| 263 |
|
| 264 |
# Assuming df is your Polars DataFrame
|
|
|
|
| 229 |
# Check if 'trajectory_null' column exists and drop it
|
| 230 |
if 'trajectory_null' in df.columns:
|
| 231 |
df = df.drop('trajectory_null')
|
| 232 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 233 |
return df
|
| 234 |
|
| 235 |
# Assuming df is your Polars DataFrame
|