Spaces:
Runtime error
Runtime error
Upload 2 files
Browse files- batting_update.py +632 -0
- pitcher_update.py +573 -0
batting_update.py
ADDED
|
@@ -0,0 +1,632 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import numpy as np
|
| 3 |
+
import joblib
|
| 4 |
+
import math
|
| 5 |
+
import pickle
|
| 6 |
+
|
| 7 |
+
loaded_model = joblib.load('joblib_model/barrel_model.joblib')
|
| 8 |
+
in_zone_model = joblib.load('joblib_model/in_zone_model_knn_20240410.joblib')
|
| 9 |
+
attack_zone_model = joblib.load('joblib_model/model_attack_zone.joblib')
|
| 10 |
+
xwoba_model = joblib.load('joblib_model/xwoba_model.joblib')
|
| 11 |
+
px_model = joblib.load('joblib_model/linear_reg_model_x.joblib')
|
| 12 |
+
pz_model = joblib.load('joblib_model/linear_reg_model_z.joblib')
|
| 13 |
+
barrel_model = joblib.load('joblib_model/barrel_model.joblib')
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
def percentile(n):
|
| 17 |
+
def percentile_(x):
|
| 18 |
+
return np.nanpercentile(x, n)
|
| 19 |
+
percentile_.__name__ = 'percentile_%s' % n
|
| 20 |
+
return percentile_
|
| 21 |
+
|
| 22 |
+
|
| 23 |
+
def df_update(df=pd.DataFrame()):
|
| 24 |
+
df.loc[df['sz_top']==0,'sz_top'] = np.nan
|
| 25 |
+
df.loc[df['sz_bot']==0,'sz_bot'] = np.nan
|
| 26 |
+
|
| 27 |
+
|
| 28 |
+
df['in_zone'] = [x < 10 if x > 0 else np.nan for x in df['zone']]
|
| 29 |
+
if len(df.loc[(~df['x'].isnull())&(df['px'].isnull()),'px']) > 0:
|
| 30 |
+
df.loc[(~df['x'].isnull())&(df['px'].isnull()),'px'] = px_model.predict(df.loc[(~df['x'].isnull())&(df['px'].isnull())][['x']])
|
| 31 |
+
df.loc[(~df['y'].isnull())&(df['pz'].isnull()),'pz'] = px_model.predict(df.loc[(~df['y'].isnull())&(df['pz'].isnull())][['y']]) + 3.2
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
# df['in_zone'] = [x < 10 if x > 0 else np.nan for x in df['zone']]
|
| 35 |
+
# df_a['in_zone'] = [x < 10 if x > 0 else np.nan for x in df_a['zone']]
|
| 36 |
+
if len(df.loc[(~df['px'].isna())&
|
| 37 |
+
(df['in_zone'].isna())&
|
| 38 |
+
(~df['sz_top'].isna())]) > 0:
|
| 39 |
+
print('We found missing data')
|
| 40 |
+
df.loc[(~df['px'].isna())&
|
| 41 |
+
(df['in_zone'].isna())&
|
| 42 |
+
(~df['sz_top'].isna())&
|
| 43 |
+
(~df['pz'].isna())&
|
| 44 |
+
(~df['sz_bot'].isna())
|
| 45 |
+
,'in_zone'] = in_zone_model.predict(df.loc[(~df['px'].isna())&
|
| 46 |
+
(df['in_zone'].isna())&
|
| 47 |
+
(~df['sz_top'].isna())&
|
| 48 |
+
(~df['pz'].isna())&
|
| 49 |
+
(~df['sz_bot'].isna())][['px','pz','sz_top','sz_bot']].values)
|
| 50 |
+
hit_codes = ['single',
|
| 51 |
+
'double','home_run', 'triple']
|
| 52 |
+
|
| 53 |
+
ab_codes = ['single', 'strikeout', 'field_out',
|
| 54 |
+
'grounded_into_double_play', 'fielders_choice', 'force_out',
|
| 55 |
+
'double', 'field_error', 'home_run', 'triple',
|
| 56 |
+
'double_play',
|
| 57 |
+
'fielders_choice_out', 'strikeout_double_play',
|
| 58 |
+
'other_out','triple_play']
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
obp_true_codes = ['single', 'walk',
|
| 62 |
+
'double','home_run', 'triple',
|
| 63 |
+
'hit_by_pitch', 'intent_walk']
|
| 64 |
+
|
| 65 |
+
obp_codes = ['single', 'strikeout', 'walk', 'field_out',
|
| 66 |
+
'grounded_into_double_play', 'fielders_choice', 'force_out',
|
| 67 |
+
'double', 'sac_fly', 'field_error', 'home_run', 'triple',
|
| 68 |
+
'hit_by_pitch', 'double_play', 'intent_walk',
|
| 69 |
+
'fielders_choice_out', 'strikeout_double_play',
|
| 70 |
+
'sac_fly_double_play',
|
| 71 |
+
'other_out','triple_play']
|
| 72 |
+
|
| 73 |
+
|
| 74 |
+
contact_codes = ['In play, no out',
|
| 75 |
+
'Foul', 'In play, out(s)',
|
| 76 |
+
'In play, run(s)',
|
| 77 |
+
'Foul Bunt']
|
| 78 |
+
|
| 79 |
+
|
| 80 |
+
|
| 81 |
+
conditions_hit = [df.event_type.isin(hit_codes)]
|
| 82 |
+
choices_hit = [True]
|
| 83 |
+
df['hits'] = np.select(conditions_hit, choices_hit, default=False)
|
| 84 |
+
|
| 85 |
+
conditions_ab = [df.event_type.isin(ab_codes)]
|
| 86 |
+
choices_ab = [True]
|
| 87 |
+
df['ab'] = np.select(conditions_ab, choices_ab, default=False)
|
| 88 |
+
|
| 89 |
+
conditions_obp_true = [df.event_type.isin(obp_true_codes)]
|
| 90 |
+
choices_obp_true = [True]
|
| 91 |
+
df['on_base'] = np.select(conditions_obp_true, choices_obp_true, default=False)
|
| 92 |
+
|
| 93 |
+
conditions_obp = [df.event_type.isin(obp_codes)]
|
| 94 |
+
choices_obp = [True]
|
| 95 |
+
df['obp'] = np.select(conditions_obp, choices_obp, default=False)
|
| 96 |
+
|
| 97 |
+
bip_codes = ['In play, no out', 'In play, run(s)','In play, out(s)']
|
| 98 |
+
|
| 99 |
+
conditions_bip = [df.play_description.isin(bip_codes)]
|
| 100 |
+
choices_bip = [True]
|
| 101 |
+
df['bip'] = np.select(conditions_bip, choices_bip, default=False)
|
| 102 |
+
|
| 103 |
+
# conditions = [
|
| 104 |
+
# (df['launch_speed'].isna()),
|
| 105 |
+
# (df['launch_speed']*1.5 - df['launch_angle'] >= 117 ) & (df['launch_speed'] + df['launch_angle'] >= 124) & (df['launch_speed'] > 98) & (df['launch_angle'] >= 8) & (df['launch_angle'] <= 50)
|
| 106 |
+
# ]
|
| 107 |
+
df['bip_div'] = ~df.launch_speed.isna()
|
| 108 |
+
# choices = [False,True]
|
| 109 |
+
# df['barrel'] = np.select(conditions, choices, default=np.nan)
|
| 110 |
+
# df['barrel'] = loaded_model.predict(df[['launch_speed','launch_angle']].fillna(0).values)
|
| 111 |
+
df['barrel'] = np.nan
|
| 112 |
+
if len(df.loc[(~df['launch_speed'].isnull())]) > 0:
|
| 113 |
+
df.loc[(~df['launch_speed'].isnull())&(~df['launch_angle'].isnull()),'barrel'] = barrel_model.predict(df.loc[(~df['launch_speed'].isnull())&(~df['launch_angle'].isnull())][['launch_speed','launch_angle']])
|
| 114 |
+
|
| 115 |
+
|
| 116 |
+
conditions_ss = [
|
| 117 |
+
(df['launch_angle'].isna()),
|
| 118 |
+
(df['launch_angle'] >= 8 ) * (df['launch_angle'] <= 32 )
|
| 119 |
+
]
|
| 120 |
+
|
| 121 |
+
choices_ss = [False,True]
|
| 122 |
+
df['sweet_spot'] = np.select(conditions_ss, choices_ss, default=np.nan)
|
| 123 |
+
|
| 124 |
+
conditions_hh = [
|
| 125 |
+
(df['launch_speed'].isna()),
|
| 126 |
+
(df['launch_speed'] >= 94.5 )
|
| 127 |
+
]
|
| 128 |
+
|
| 129 |
+
choices_hh = [False,True]
|
| 130 |
+
df['hard_hit'] = np.select(conditions_hh, choices_hh, default=np.nan)
|
| 131 |
+
|
| 132 |
+
|
| 133 |
+
conditions_tb = [
|
| 134 |
+
(df['event_type']=='single'),
|
| 135 |
+
(df['event_type']=='double'),
|
| 136 |
+
(df['event_type']=='triple'),
|
| 137 |
+
(df['event_type']=='home_run'),
|
| 138 |
+
]
|
| 139 |
+
|
| 140 |
+
choices_tb = [1,2,3,4]
|
| 141 |
+
|
| 142 |
+
df['tb'] = np.select(conditions_tb, choices_tb, default=np.nan)
|
| 143 |
+
|
| 144 |
+
conditions_woba = [
|
| 145 |
+
(df['event_type'].isin(['strikeout', 'field_out', 'sac_fly', 'force_out',
|
| 146 |
+
'grounded_into_double_play', 'fielders_choice', 'field_error',
|
| 147 |
+
'sac_bunt', 'double_play', 'fielders_choice_out', 'strikeout_double_play',
|
| 148 |
+
'sac_fly_double_play', 'other_out'])),
|
| 149 |
+
(df['event_type']=='walk'),
|
| 150 |
+
(df['event_type']=='hit_by_pitch'),
|
| 151 |
+
(df['event_type']=='single'),
|
| 152 |
+
(df['event_type']=='double'),
|
| 153 |
+
(df['event_type']=='triple'),
|
| 154 |
+
(df['event_type']=='home_run'),
|
| 155 |
+
]
|
| 156 |
+
|
| 157 |
+
choices_woba = [0,
|
| 158 |
+
0.696,
|
| 159 |
+
0.726,
|
| 160 |
+
0.883,
|
| 161 |
+
1.244,
|
| 162 |
+
1.569,
|
| 163 |
+
2.004]
|
| 164 |
+
|
| 165 |
+
df['woba'] = np.select(conditions_woba, choices_woba, default=np.nan)
|
| 166 |
+
|
| 167 |
+
|
| 168 |
+
woba_codes = ['strikeout', 'field_out', 'single', 'walk', 'hit_by_pitch',
|
| 169 |
+
'double', 'sac_fly', 'force_out', 'home_run',
|
| 170 |
+
'grounded_into_double_play', 'fielders_choice', 'field_error',
|
| 171 |
+
'triple', 'sac_bunt', 'double_play',
|
| 172 |
+
'fielders_choice_out', 'strikeout_double_play',
|
| 173 |
+
'sac_fly_double_play', 'other_out']
|
| 174 |
+
|
| 175 |
+
|
| 176 |
+
|
| 177 |
+
|
| 178 |
+
|
| 179 |
+
|
| 180 |
+
conditions_woba_code = [
|
| 181 |
+
(df['event_type'].isin(woba_codes))
|
| 182 |
+
]
|
| 183 |
+
|
| 184 |
+
choices_woba_code = [1]
|
| 185 |
+
|
| 186 |
+
df['woba_codes'] = np.select(conditions_woba_code, choices_woba_code, default=np.nan)
|
| 187 |
+
|
| 188 |
+
|
| 189 |
+
df['woba_contact'] = [df['woba'].values[x] if df['bip'].values[x] == 1 else np.nan for x in range(len(df['woba_codes']))]
|
| 190 |
+
|
| 191 |
+
#df['in_zone'] = [x < 10 if type(x) == int else np.nan for x in df['zone']]
|
| 192 |
+
|
| 193 |
+
# df['in_zone_2'] = in_zone_model.predict(df[['x','y','sz_bot','sz_top']].fillna(0).values)
|
| 194 |
+
# df['in_zone_3'] = df['in_zone_2'] < 10
|
| 195 |
+
# df.loc[df['in_zone'].isna(),'in_zone'] = df.loc[df['in_zone'].isna(),'in_zone_3'].fillna(0)
|
| 196 |
+
|
| 197 |
+
|
| 198 |
+
df['whiffs'] = [1 if ((x == 'S')|(x == 'W')|(x =='T')) else 0 for x in df.play_code]
|
| 199 |
+
df['csw'] = [1 if ((x == 'S')|(x == 'W')|(x =='T')|(x == 'C')) else 0 for x in df.play_code]
|
| 200 |
+
df['swings'] = [1 if x == True else 0 for x in df.is_swing]
|
| 201 |
+
|
| 202 |
+
|
| 203 |
+
df['out_zone'] = df.in_zone == False
|
| 204 |
+
df['zone_swing'] = (df.in_zone == True)&(df.swings == 1)
|
| 205 |
+
df['zone_contact'] = (df.in_zone == True)&(df.swings == 1)&(df.whiffs == 0)
|
| 206 |
+
df['ozone_swing'] = (df.in_zone==False)&(df.swings == 1)
|
| 207 |
+
df['ozone_contact'] = (df.in_zone==False)&(df.swings == 1)&(df.whiffs == 0)
|
| 208 |
+
|
| 209 |
+
df['k'] = df.event_type.isin(list(filter(None, [x if 'strikeout' in x else '' for x in df.event_type.dropna().unique()])))
|
| 210 |
+
df['bb'] = df.event_type.isin(['walk','intent_walk'])
|
| 211 |
+
|
| 212 |
+
df['k_minus_bb'] = df['k'].astype(np.float32)-df['bb'].astype(np.float32)
|
| 213 |
+
df['bb_minus_k'] = df['bb'].astype(np.float32)-df['k'].astype(np.float32)
|
| 214 |
+
|
| 215 |
+
df['pa'] = [1 if isinstance(x, str) else 0 for x in df.event_type]
|
| 216 |
+
df['pitches'] = [1 if x else 0 for x in df.is_pitch]
|
| 217 |
+
|
| 218 |
+
|
| 219 |
+
df.loc[df['launch_speed'].isna(),'barrel'] = np.nan
|
| 220 |
+
|
| 221 |
+
|
| 222 |
+
pitch_cat = {'FA':'Fastball',
|
| 223 |
+
'FF':'Fastball',
|
| 224 |
+
'FT':'Fastball',
|
| 225 |
+
'FC':'Fastball',
|
| 226 |
+
'FS':'Off-Speed',
|
| 227 |
+
'FO':'Off-Speed',
|
| 228 |
+
'SI':'Fastball',
|
| 229 |
+
'ST':'Breaking',
|
| 230 |
+
'SL':'Breaking',
|
| 231 |
+
'CU':'Breaking',
|
| 232 |
+
'KC':'Breaking',
|
| 233 |
+
'SC':'Off-Speed',
|
| 234 |
+
'GY':'Off-Speed',
|
| 235 |
+
'SV':'Breaking',
|
| 236 |
+
'CS':'Breaking',
|
| 237 |
+
'CH':'Off-Speed',
|
| 238 |
+
'KN':'Off-Speed',
|
| 239 |
+
'EP':'Breaking',
|
| 240 |
+
'UN':np.nan,
|
| 241 |
+
'IN':np.nan,
|
| 242 |
+
'PO':np.nan,
|
| 243 |
+
'AB':np.nan,
|
| 244 |
+
'AS':np.nan,
|
| 245 |
+
'NP':np.nan}
|
| 246 |
+
df['pitch_category'] = df['pitch_type'].map(pitch_cat).fillna('Unknown')
|
| 247 |
+
df['average'] = 'average'
|
| 248 |
+
|
| 249 |
+
df.loc[df['trajectory'] == 'bunt_popup','trajectory'] = 'popup'
|
| 250 |
+
df.loc[df['trajectory'] == 'bunt_grounder','trajectory'] = 'ground_ball'
|
| 251 |
+
df.loc[df['trajectory'] == '','trajectory'] = np.nan
|
| 252 |
+
df.loc[df['trajectory'] == 'bunt_line_drive','trajectory'] = 'line_drive'
|
| 253 |
+
df[['trajectory_fly_ball','trajectory_ground_ball','trajectory_line_drive','trajectory_popup']] = pd.get_dummies(df['trajectory'], prefix='trajectory')
|
| 254 |
+
|
| 255 |
+
df['attack_zone'] = np.nan
|
| 256 |
+
|
| 257 |
+
|
| 258 |
+
|
| 259 |
+
df.loc[df[['px','pz','sz_top','sz_bot']].isnull().sum(axis=1)==0,'attack_zone'] = attack_zone_model.predict(df.loc[df[['px','pz','sz_top','sz_bot']].isnull().sum(axis=1)==0][['px','pz','sz_top','sz_bot']])
|
| 260 |
+
|
| 261 |
+
|
| 262 |
+
|
| 263 |
+
df['heart'] = df['attack_zone'] == 0
|
| 264 |
+
df['shadow'] = df['attack_zone'] == 1
|
| 265 |
+
df['chase'] = df['attack_zone'] == 2
|
| 266 |
+
df['waste'] = df['attack_zone'] == 3
|
| 267 |
+
|
| 268 |
+
df['heart_swing'] = (df['attack_zone'] == 0)&(df['swings']==1)
|
| 269 |
+
df['shadow_swing'] = (df['attack_zone'] == 1)&(df['swings']==1)
|
| 270 |
+
df['chase_swing'] = (df['attack_zone'] == 2)&(df['swings']==1)
|
| 271 |
+
df['waste_swing'] = (df['attack_zone'] == 3)&(df['swings']==1)
|
| 272 |
+
|
| 273 |
+
df['xwoba'] = np.nan
|
| 274 |
+
df['xwoba_contact'] = np.nan
|
| 275 |
+
|
| 276 |
+
if len(df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'xwoba']) > 0:
|
| 277 |
+
|
| 278 |
+
|
| 279 |
+
df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'xwoba'] = [sum(x) for x in xwoba_model.predict_proba(df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0][['launch_angle','launch_speed']]) * ([0, 0.883,1.244,1.569,2.004])]
|
| 280 |
+
|
| 281 |
+
## Assign a value of 0.696 to every walk in the dataset
|
| 282 |
+
df.loc[df['event_type'].isin(['walk']),'xwoba'] = 0.696
|
| 283 |
+
|
| 284 |
+
## Assign a value of 0.726 to every hit by pitch in the dataset
|
| 285 |
+
df.loc[df['event_type'].isin(['hit_by_pitch']),'xwoba'] = 0.726
|
| 286 |
+
|
| 287 |
+
## Assign a value of 0 to every Strikeout in the dataset
|
| 288 |
+
df.loc[df['event_type'].isin(['strikeout','strikeout_double_play']),'xwoba'] = 0
|
| 289 |
+
|
| 290 |
+
|
| 291 |
+
df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'xwoba_contact'] = [sum(x) for x in xwoba_model.predict_proba(df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0][['launch_angle','launch_speed']]) * ([0, 0.883,1.244,1.569,2.004])]
|
| 292 |
+
|
| 293 |
+
df['xwoba_codes'] = np.nan
|
| 294 |
+
df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'xwoba_codes'] = 1
|
| 295 |
+
## Assign a value of 0.696 to every walk in the dataset
|
| 296 |
+
df.loc[df['event_type'].isin(['walk']),'xwoba_codes'] = 1
|
| 297 |
+
|
| 298 |
+
## Assign a value of 0.726 to every hit by pitch in the dataset
|
| 299 |
+
df.loc[df['event_type'].isin(['hit_by_pitch']),'xwoba_codes'] = 1
|
| 300 |
+
|
| 301 |
+
## Assign a value of 0 to every Strikeout in the dataset
|
| 302 |
+
df.loc[df['event_type'].isin(['strikeout','strikeout_double_play']),'xwoba_codes'] = 1
|
| 303 |
+
return df
|
| 304 |
+
|
| 305 |
+
def df_update_summ(df=pd.DataFrame()):
|
| 306 |
+
df_summ = df.groupby(['batter_id','batter_name']).agg(
|
| 307 |
+
pa = ('pa','sum'),
|
| 308 |
+
ab = ('ab','sum'),
|
| 309 |
+
obp_pa = ('obp','sum'),
|
| 310 |
+
hits = ('hits','sum'),
|
| 311 |
+
on_base = ('on_base','sum'),
|
| 312 |
+
k = ('k','sum'),
|
| 313 |
+
bb = ('bb','sum'),
|
| 314 |
+
bb_minus_k = ('bb_minus_k','sum'),
|
| 315 |
+
csw = ('csw','sum'),
|
| 316 |
+
bip = ('bip','sum'),
|
| 317 |
+
bip_div = ('bip_div','sum'),
|
| 318 |
+
tb = ('tb','sum'),
|
| 319 |
+
woba = ('woba','sum'),
|
| 320 |
+
woba_contact = ('woba_contact','sum'),
|
| 321 |
+
xwoba = ('xwoba','sum'),
|
| 322 |
+
xwoba_contact = ('xwoba_contact','sum'),
|
| 323 |
+
woba_codes = ('woba_codes','sum'),
|
| 324 |
+
xwoba_codes = ('xwoba_codes','sum'),
|
| 325 |
+
hard_hit = ('hard_hit','sum'),
|
| 326 |
+
barrel = ('barrel','sum'),
|
| 327 |
+
sweet_spot = ('sweet_spot','sum'),
|
| 328 |
+
max_launch_speed = ('launch_speed','max'),
|
| 329 |
+
launch_speed_90 = ('launch_speed',percentile(90)),
|
| 330 |
+
launch_speed = ('launch_speed','mean'),
|
| 331 |
+
launch_angle = ('launch_angle','mean'),
|
| 332 |
+
pitches = ('is_pitch','sum'),
|
| 333 |
+
swings = ('swings','sum'),
|
| 334 |
+
in_zone = ('in_zone','sum'),
|
| 335 |
+
out_zone = ('out_zone','sum'),
|
| 336 |
+
whiffs = ('whiffs','sum'),
|
| 337 |
+
zone_swing = ('zone_swing','sum'),
|
| 338 |
+
zone_contact = ('zone_contact','sum'),
|
| 339 |
+
ozone_swing = ('ozone_swing','sum'),
|
| 340 |
+
ozone_contact = ('ozone_contact','sum'),
|
| 341 |
+
ground_ball = ('trajectory_ground_ball','sum'),
|
| 342 |
+
line_drive = ('trajectory_line_drive','sum'),
|
| 343 |
+
fly_ball =('trajectory_fly_ball','sum'),
|
| 344 |
+
pop_up = ('trajectory_popup','sum'),
|
| 345 |
+
attack_zone = ('attack_zone','count'),
|
| 346 |
+
heart = ('heart','sum'),
|
| 347 |
+
shadow = ('shadow','sum'),
|
| 348 |
+
chase = ('chase','sum'),
|
| 349 |
+
waste = ('waste','sum'),
|
| 350 |
+
heart_swing = ('heart_swing','sum'),
|
| 351 |
+
shadow_swing = ('shadow_swing','sum'),
|
| 352 |
+
chase_swing = ('chase_swing','sum'),
|
| 353 |
+
waste_swing = ('waste_swing','sum'),
|
| 354 |
+
).reset_index()
|
| 355 |
+
return df_summ
|
| 356 |
+
|
| 357 |
+
def df_update_summ_avg(df=pd.DataFrame()):
|
| 358 |
+
df_summ_avg = df.groupby(['average']).agg(
|
| 359 |
+
pa = ('pa','sum'),
|
| 360 |
+
ab = ('ab','sum'),
|
| 361 |
+
obp_pa = ('obp','sum'),
|
| 362 |
+
hits = ('hits','sum'),
|
| 363 |
+
on_base = ('on_base','sum'),
|
| 364 |
+
k = ('k','sum'),
|
| 365 |
+
bb = ('bb','sum'),
|
| 366 |
+
bb_minus_k = ('bb_minus_k','sum'),
|
| 367 |
+
csw = ('csw','sum'),
|
| 368 |
+
bip = ('bip','sum'),
|
| 369 |
+
bip_div = ('bip_div','sum'),
|
| 370 |
+
tb = ('tb','sum'),
|
| 371 |
+
woba = ('woba','sum'),
|
| 372 |
+
woba_contact = ('woba_contact','sum'),
|
| 373 |
+
xwoba = ('xwoba','sum'),
|
| 374 |
+
xwoba_contact = ('xwoba_contact','sum'),
|
| 375 |
+
woba_codes = ('woba_codes','sum'),
|
| 376 |
+
xwoba_codes = ('xwoba_codes','sum'),
|
| 377 |
+
hard_hit = ('hard_hit','sum'),
|
| 378 |
+
barrel = ('barrel','sum'),
|
| 379 |
+
sweet_spot = ('sweet_spot','sum'),
|
| 380 |
+
max_launch_speed = ('launch_speed','max'),
|
| 381 |
+
launch_speed_90 = ('launch_speed',percentile(90)),
|
| 382 |
+
launch_speed = ('launch_speed','mean'),
|
| 383 |
+
launch_angle = ('launch_angle','mean'),
|
| 384 |
+
pitches = ('is_pitch','sum'),
|
| 385 |
+
swings = ('swings','sum'),
|
| 386 |
+
in_zone = ('in_zone','sum'),
|
| 387 |
+
out_zone = ('out_zone','sum'),
|
| 388 |
+
whiffs = ('whiffs','sum'),
|
| 389 |
+
zone_swing = ('zone_swing','sum'),
|
| 390 |
+
zone_contact = ('zone_contact','sum'),
|
| 391 |
+
ozone_swing = ('ozone_swing','sum'),
|
| 392 |
+
ozone_contact = ('ozone_contact','sum'),
|
| 393 |
+
ground_ball = ('trajectory_ground_ball','sum'),
|
| 394 |
+
line_drive = ('trajectory_line_drive','sum'),
|
| 395 |
+
fly_ball =('trajectory_fly_ball','sum'),
|
| 396 |
+
pop_up = ('trajectory_popup','sum'),
|
| 397 |
+
attack_zone = ('attack_zone','count'),
|
| 398 |
+
heart = ('heart','sum'),
|
| 399 |
+
shadow = ('shadow','sum'),
|
| 400 |
+
chase = ('chase','sum'),
|
| 401 |
+
waste = ('waste','sum'),
|
| 402 |
+
heart_swing = ('heart_swing','sum'),
|
| 403 |
+
shadow_swing = ('shadow_swing','sum'),
|
| 404 |
+
chase_swing = ('chase_swing','sum'),
|
| 405 |
+
waste_swing = ('waste_swing','sum'),
|
| 406 |
+
|
| 407 |
+
|
| 408 |
+
|
| 409 |
+
|
| 410 |
+
).reset_index()
|
| 411 |
+
return df_summ_avg
|
| 412 |
+
|
| 413 |
+
def df_summ_changes(df_summ=pd.DataFrame()):
|
| 414 |
+
df_summ['avg'] = [df_summ.hits[x]/df_summ.ab[x] if df_summ.ab[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 415 |
+
df_summ['obp'] = [df_summ.on_base[x]/df_summ.obp_pa[x] if df_summ.obp_pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 416 |
+
df_summ['slg'] = [df_summ.tb[x]/df_summ.ab[x] if df_summ.ab[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 417 |
+
|
| 418 |
+
df_summ['ops'] = df_summ['obp']+df_summ['slg']
|
| 419 |
+
|
| 420 |
+
df_summ['k_percent'] = [df_summ.k[x]/df_summ.pa[x] if df_summ.pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 421 |
+
df_summ['bb_percent'] =[df_summ.bb[x]/df_summ.pa[x] if df_summ.pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 422 |
+
df_summ['bb_minus_k_percent'] =[(df_summ.bb_minus_k[x])/df_summ.pa[x] if df_summ.pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 423 |
+
|
| 424 |
+
df_summ['bb_over_k_percent'] =[df_summ.bb[x]/df_summ.k[x] if df_summ.k[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 425 |
+
|
| 426 |
+
|
| 427 |
+
|
| 428 |
+
|
| 429 |
+
df_summ['csw_percent'] =[df_summ.csw[x]/df_summ.pitches[x] if df_summ.pitches[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 430 |
+
|
| 431 |
+
|
| 432 |
+
df_summ['sweet_spot_percent'] = [df_summ.sweet_spot[x]/df_summ.bip_div[x] if df_summ.bip_div[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 433 |
+
|
| 434 |
+
df_summ['woba_percent'] = [df_summ.woba[x]/df_summ.woba_codes[x] if df_summ.woba_codes[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 435 |
+
df_summ['woba_percent_contact'] = [df_summ.woba_contact[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 436 |
+
#df_summ['hard_hit_percent'] = [df_summ.sweet_spot[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 437 |
+
df_summ['hard_hit_percent'] = [df_summ.hard_hit[x]/df_summ.bip_div[x] if df_summ.bip_div[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 438 |
+
|
| 439 |
+
|
| 440 |
+
df_summ['barrel_percent'] = [df_summ.barrel[x]/df_summ.bip_div[x] if df_summ.bip_div[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 441 |
+
|
| 442 |
+
df_summ['zone_contact_percent'] = [df_summ.zone_contact[x]/df_summ.zone_swing[x] if df_summ.zone_swing[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 443 |
+
|
| 444 |
+
df_summ['zone_swing_percent'] = [df_summ.zone_swing[x]/df_summ.in_zone[x] if df_summ.in_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 445 |
+
|
| 446 |
+
df_summ['zone_percent'] = [df_summ.in_zone[x]/df_summ.pitches[x] if df_summ.pitches[x] > 0 else np.nan for x in range(len(df_summ))]
|
| 447 |
+
|
| 448 |
+
df_summ['chase_percent'] = [df_summ.ozone_swing[x]/(df_summ.pitches[x] - df_summ.in_zone[x]) if (df_summ.pitches[x]- df_summ.in_zone[x]) != 0 else np.nan for x in range(len(df_summ))]
|
| 449 |
+
|
| 450 |
+
df_summ['chase_contact'] = [df_summ.ozone_contact[x]/df_summ.ozone_swing[x] if df_summ.ozone_swing[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 451 |
+
|
| 452 |
+
df_summ['swing_percent'] = [df_summ.swings[x]/df_summ.pitches[x] if df_summ.pitches[x] > 0 else np.nan for x in range(len(df_summ))]
|
| 453 |
+
|
| 454 |
+
df_summ['whiff_rate'] = [df_summ.whiffs[x]/df_summ.swings[x] if df_summ.swings[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 455 |
+
|
| 456 |
+
df_summ['swstr_rate'] = [df_summ.whiffs[x]/df_summ.pitches[x] if df_summ.pitches[x] > 0 else np.nan for x in range(len(df_summ))]
|
| 457 |
+
|
| 458 |
+
df_summ['ground_ball_percent'] = [df_summ.ground_ball[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 459 |
+
|
| 460 |
+
df_summ['line_drive_percent'] = [df_summ.line_drive[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 461 |
+
|
| 462 |
+
df_summ['fly_ball_percent'] = [df_summ.fly_ball[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 463 |
+
|
| 464 |
+
df_summ['pop_up_percent'] = [df_summ.pop_up[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 465 |
+
|
| 466 |
+
|
| 467 |
+
|
| 468 |
+
df_summ['heart_zone_percent'] = [df_summ.heart[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 469 |
+
|
| 470 |
+
df_summ['shadow_zone_percent'] = [df_summ.shadow[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 471 |
+
|
| 472 |
+
df_summ['chase_zone_percent'] = [df_summ.chase[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 473 |
+
|
| 474 |
+
df_summ['waste_zone_percent'] = [df_summ.waste[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 475 |
+
|
| 476 |
+
|
| 477 |
+
df_summ['heart_zone_swing_percent'] = [df_summ.heart_swing[x]/df_summ.heart[x] if df_summ.heart[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 478 |
+
|
| 479 |
+
df_summ['shadow_zone_swing_percent'] = [df_summ.shadow_swing[x]/df_summ.shadow[x] if df_summ.shadow[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 480 |
+
|
| 481 |
+
df_summ['chase_zone_swing_percent'] = [df_summ.chase_swing[x]/df_summ.chase[x] if df_summ.chase[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 482 |
+
|
| 483 |
+
df_summ['waste_zone_swing_percent'] = [df_summ.waste_swing[x]/df_summ.waste[x] if df_summ.waste[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 484 |
+
|
| 485 |
+
|
| 486 |
+
df_summ['xwoba_percent'] = [df_summ.xwoba[x]/df_summ.xwoba_codes[x] if df_summ.xwoba_codes[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 487 |
+
df_summ['xwoba_percent_contact'] = [df_summ.xwoba_contact[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 488 |
+
|
| 489 |
+
df_summ = df_summ.dropna(subset=['bip'])
|
| 490 |
+
return df_summ
|
| 491 |
+
|
| 492 |
+
def df_summ_filter_out(df_summ=pd.DataFrame(),batter_select = 0,date_min=0):
|
| 493 |
+
import datetime
|
| 494 |
+
|
| 495 |
+
def weeks_after(day):
|
| 496 |
+
today = datetime.date.today()
|
| 497 |
+
time_difference = today - day
|
| 498 |
+
weeks = time_difference.days // 7
|
| 499 |
+
return weeks
|
| 500 |
+
|
| 501 |
+
df_summ_filter = df_summ[df_summ['pa'] >= min(math.floor(df_summ.xs(batter_select,level=0)['pa']/10)*10,500,weeks_after(date_min)*20)]
|
| 502 |
+
df_summ_filter_pct = df_summ_filter.rank(pct=True,ascending=True)
|
| 503 |
+
df_summ_player = df_summ.xs(batter_select,level=0)
|
| 504 |
+
df_summ_player_pct = df_summ_filter_pct.xs(batter_select,level=0)
|
| 505 |
+
return df_summ_filter,df_summ_filter_pct,df_summ_player,df_summ_player_pct
|
| 506 |
+
|
| 507 |
+
|
| 508 |
+
def df_summ_batter_pitch_up(df=pd.DataFrame()):
|
| 509 |
+
df_summ_batter_pitch = df.dropna(subset=['pitch_category']).groupby(['batter_id','batter_name','pitch_category']).agg(
|
| 510 |
+
pa = ('pa','sum'),
|
| 511 |
+
ab = ('ab','sum'),
|
| 512 |
+
obp_pa = ('obp','sum'),
|
| 513 |
+
hits = ('hits','sum'),
|
| 514 |
+
on_base = ('on_base','sum'),
|
| 515 |
+
k = ('k','sum'),
|
| 516 |
+
bb = ('bb','sum'),
|
| 517 |
+
bb_minus_k = ('bb_minus_k','sum'),
|
| 518 |
+
csw = ('csw','sum'),
|
| 519 |
+
bip = ('bip','sum'),
|
| 520 |
+
bip_div = ('bip_div','sum'),
|
| 521 |
+
tb = ('tb','sum'),
|
| 522 |
+
woba = ('woba','sum'),
|
| 523 |
+
woba_contact = ('xwoba_contact','sum'),
|
| 524 |
+
xwoba = ('xwoba','sum'),
|
| 525 |
+
xwoba_contact = ('xwoba','sum'),
|
| 526 |
+
woba_codes = ('woba_codes','sum'),
|
| 527 |
+
xwoba_codes = ('xwoba_codes','sum'),
|
| 528 |
+
hard_hit = ('hard_hit','sum'),
|
| 529 |
+
barrel = ('barrel','sum'),
|
| 530 |
+
sweet_spot = ('sweet_spot','sum'),
|
| 531 |
+
max_launch_speed = ('launch_speed','max'),
|
| 532 |
+
launch_speed_90 = ('launch_speed',percentile(90)),
|
| 533 |
+
launch_speed = ('launch_speed','mean'),
|
| 534 |
+
launch_angle = ('launch_angle','mean'),
|
| 535 |
+
pitches = ('is_pitch','sum'),
|
| 536 |
+
swings = ('swings','sum'),
|
| 537 |
+
in_zone = ('in_zone','sum'),
|
| 538 |
+
out_zone = ('out_zone','sum'),
|
| 539 |
+
whiffs = ('whiffs','sum'),
|
| 540 |
+
zone_swing = ('zone_swing','sum'),
|
| 541 |
+
zone_contact = ('zone_contact','sum'),
|
| 542 |
+
ozone_swing = ('ozone_swing','sum'),
|
| 543 |
+
ozone_contact = ('ozone_contact','sum'),
|
| 544 |
+
ground_ball = ('trajectory_ground_ball','sum'),
|
| 545 |
+
line_drive = ('trajectory_line_drive','sum'),
|
| 546 |
+
fly_ball =('trajectory_fly_ball','sum'),
|
| 547 |
+
pop_up = ('trajectory_popup','sum'),
|
| 548 |
+
attack_zone = ('attack_zone','count'),
|
| 549 |
+
heart = ('heart','sum'),
|
| 550 |
+
shadow = ('shadow','sum'),
|
| 551 |
+
chase = ('chase','sum'),
|
| 552 |
+
waste = ('waste','sum'),
|
| 553 |
+
heart_swing = ('heart_swing','sum'),
|
| 554 |
+
shadow_swing = ('shadow_swing','sum'),
|
| 555 |
+
chase_swing = ('chase_swing','sum'),
|
| 556 |
+
waste_swing = ('waste_swing','sum'),
|
| 557 |
+
).reset_index()
|
| 558 |
+
|
| 559 |
+
#return df_summ_batter_pitch
|
| 560 |
+
df_summ_batter_pitch['avg'] = [df_summ_batter_pitch.hits[x]/df_summ_batter_pitch.ab[x] if df_summ_batter_pitch.ab[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 561 |
+
df_summ_batter_pitch['obp'] = [df_summ_batter_pitch.on_base[x]/df_summ_batter_pitch.obp_pa[x] if df_summ_batter_pitch.obp_pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 562 |
+
df_summ_batter_pitch['slg'] = [df_summ_batter_pitch.tb[x]/df_summ_batter_pitch.ab[x] if df_summ_batter_pitch.ab[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 563 |
+
|
| 564 |
+
df_summ_batter_pitch['ops'] = df_summ_batter_pitch['obp']+df_summ_batter_pitch['slg']
|
| 565 |
+
|
| 566 |
+
df_summ_batter_pitch['k_percent'] = [df_summ_batter_pitch.k[x]/df_summ_batter_pitch.pa[x] if df_summ_batter_pitch.pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 567 |
+
df_summ_batter_pitch['bb_percent'] =[df_summ_batter_pitch.bb[x]/df_summ_batter_pitch.pa[x] if df_summ_batter_pitch.pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 568 |
+
df_summ_batter_pitch['bb_minus_k_percent'] =[(df_summ_batter_pitch.bb_minus_k[x])/df_summ_batter_pitch.pa[x] if df_summ_batter_pitch.pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 569 |
+
|
| 570 |
+
df_summ_batter_pitch['bb_over_k_percent'] =[df_summ_batter_pitch.bb[x]/df_summ_batter_pitch.k[x] if df_summ_batter_pitch.k[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 571 |
+
|
| 572 |
+
|
| 573 |
+
|
| 574 |
+
|
| 575 |
+
df_summ_batter_pitch['csw_percent'] =[df_summ_batter_pitch.csw[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 576 |
+
|
| 577 |
+
|
| 578 |
+
df_summ_batter_pitch['sweet_spot_percent'] = [df_summ_batter_pitch.sweet_spot[x]/df_summ_batter_pitch.bip_div[x] if df_summ_batter_pitch.bip_div[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 579 |
+
|
| 580 |
+
df_summ_batter_pitch['woba_percent'] = [df_summ_batter_pitch.woba[x]/df_summ_batter_pitch.woba_codes[x] if df_summ_batter_pitch.woba_codes[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 581 |
+
df_summ_batter_pitch['woba_percent_contact'] = [df_summ_batter_pitch.woba_contact[x]/df_summ_batter_pitch.bip[x] if df_summ_batter_pitch.bip[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 582 |
+
#df_summ_batter_pitch['hard_hit_percent'] = [df_summ_batter_pitch.sweet_spot[x]/df_summ_batter_pitch.bip[x] if df_summ_batter_pitch.bip[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 583 |
+
df_summ_batter_pitch['hard_hit_percent'] = [df_summ_batter_pitch.hard_hit[x]/df_summ_batter_pitch.bip_div[x] if df_summ_batter_pitch.bip_div[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 584 |
+
|
| 585 |
+
|
| 586 |
+
df_summ_batter_pitch['barrel_percent'] = [df_summ_batter_pitch.barrel[x]/df_summ_batter_pitch.bip_div[x] if df_summ_batter_pitch.bip_div[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 587 |
+
|
| 588 |
+
df_summ_batter_pitch['zone_contact_percent'] = [df_summ_batter_pitch.zone_contact[x]/df_summ_batter_pitch.zone_swing[x] if df_summ_batter_pitch.zone_swing[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 589 |
+
|
| 590 |
+
df_summ_batter_pitch['zone_swing_percent'] = [df_summ_batter_pitch.zone_swing[x]/df_summ_batter_pitch.in_zone[x] if df_summ_batter_pitch.in_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 591 |
+
|
| 592 |
+
df_summ_batter_pitch['zone_percent'] = [df_summ_batter_pitch.in_zone[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 593 |
+
|
| 594 |
+
df_summ_batter_pitch['chase_percent'] = [df_summ_batter_pitch.ozone_swing[x]/(df_summ_batter_pitch.pitches[x] - df_summ_batter_pitch.in_zone[x]) if (df_summ_batter_pitch.pitches[x]- df_summ_batter_pitch.in_zone[x]) != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 595 |
+
|
| 596 |
+
df_summ_batter_pitch['chase_contact'] = [df_summ_batter_pitch.ozone_contact[x]/df_summ_batter_pitch.ozone_swing[x] if df_summ_batter_pitch.ozone_swing[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 597 |
+
|
| 598 |
+
df_summ_batter_pitch['swing_percent'] = [df_summ_batter_pitch.swings[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 599 |
+
|
| 600 |
+
df_summ_batter_pitch['whiff_rate'] = [df_summ_batter_pitch.whiffs[x]/df_summ_batter_pitch.swings[x] if df_summ_batter_pitch.swings[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 601 |
+
|
| 602 |
+
df_summ_batter_pitch['swstr_rate'] = [df_summ_batter_pitch.whiffs[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 603 |
+
|
| 604 |
+
df_summ_batter_pitch['heart_zone_percent'] = [df_summ_batter_pitch.heart[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 605 |
+
|
| 606 |
+
df_summ_batter_pitch['shadow_zone_percent'] = [df_summ_batter_pitch.shadow[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 607 |
+
|
| 608 |
+
df_summ_batter_pitch['chase_zone_percent'] = [df_summ_batter_pitch.chase[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 609 |
+
|
| 610 |
+
df_summ_batter_pitch['waste_zone_percent'] = [df_summ_batter_pitch.waste[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 611 |
+
|
| 612 |
+
|
| 613 |
+
df_summ_batter_pitch['heart_zone_swing_percent'] = [df_summ_batter_pitch.heart_swing[x]/df_summ_batter_pitch.heart[x] if df_summ_batter_pitch.heart[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 614 |
+
|
| 615 |
+
df_summ_batter_pitch['shadow_zone_swing_percent'] = [df_summ_batter_pitch.shadow_swing[x]/df_summ_batter_pitch.shadow[x] if df_summ_batter_pitch.shadow[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 616 |
+
|
| 617 |
+
df_summ_batter_pitch['chase_zone_swing_percent'] = [df_summ_batter_pitch.chase_swing[x]/df_summ_batter_pitch.chase[x] if df_summ_batter_pitch.chase[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 618 |
+
|
| 619 |
+
df_summ_batter_pitch['waste_zone_swing_percent'] = [df_summ_batter_pitch.waste_swing[x]/df_summ_batter_pitch.waste[x] if df_summ_batter_pitch.waste[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 620 |
+
|
| 621 |
+
|
| 622 |
+
|
| 623 |
+
|
| 624 |
+
df_summ_batter_pitch['xwoba_percent'] = [df_summ_batter_pitch.xwoba[x]/df_summ_batter_pitch.xwoba_codes[x] if df_summ_batter_pitch.xwoba_codes[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 625 |
+
df_summ_batter_pitch['xwoba_percent_contact'] = [df_summ_batter_pitch.xwoba_contact[x]/df_summ_batter_pitch.bip[x] if df_summ_batter_pitch.bip[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 626 |
+
|
| 627 |
+
|
| 628 |
+
|
| 629 |
+
|
| 630 |
+
df_summ_batter_pitch['bip'] = df_summ_batter_pitch['bip'].fillna(0)
|
| 631 |
+
|
| 632 |
+
return df_summ_batter_pitch
|
pitcher_update.py
ADDED
|
@@ -0,0 +1,573 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import numpy as np
|
| 3 |
+
import joblib
|
| 4 |
+
import math
|
| 5 |
+
import pickle
|
| 6 |
+
|
| 7 |
+
loaded_model = joblib.load('joblib_model/barrel_model.joblib')
|
| 8 |
+
in_zone_model = joblib.load('joblib_model/in_zone_model_knn_20240410.joblib')
|
| 9 |
+
attack_zone_model = joblib.load('joblib_model/model_attack_zone.joblib')
|
| 10 |
+
xwoba_model = joblib.load('joblib_model/xwoba_model.joblib')
|
| 11 |
+
px_model = joblib.load('joblib_model/linear_reg_model_x.joblib')
|
| 12 |
+
pz_model = joblib.load('joblib_model/linear_reg_model_z.joblib')
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
def percentile(n):
|
| 16 |
+
def percentile_(x):
|
| 17 |
+
return np.nanpercentile(x, n)
|
| 18 |
+
percentile_.__name__ = 'percentile_%s' % n
|
| 19 |
+
return percentile_
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
def df_update(df=pd.DataFrame()):
|
| 23 |
+
df.loc[df['sz_top']==0,'sz_top'] = np.nan
|
| 24 |
+
df.loc[df['sz_bot']==0,'sz_bot'] = np.nan
|
| 25 |
+
|
| 26 |
+
|
| 27 |
+
df['in_zone'] = [x < 10 if x > 0 else np.nan for x in df['zone']]
|
| 28 |
+
if len(df.loc[(~df['x'].isnull())&(df['px'].isnull()),'px']) > 0:
|
| 29 |
+
df.loc[(~df['x'].isnull())&(df['px'].isnull()),'px'] = px_model.predict(df.loc[(~df['x'].isnull())&(df['px'].isnull())][['x']])
|
| 30 |
+
df.loc[(~df['y'].isnull())&(df['pz'].isnull()),'pz'] = px_model.predict(df.loc[(~df['y'].isnull())&(df['pz'].isnull())][['y']]) + 3.2
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
# df['in_zone'] = [x < 10 if x > 0 else np.nan for x in df['zone']]
|
| 34 |
+
if len(df.loc[(~df['px'].isna())&
|
| 35 |
+
(df['in_zone'].isna())&
|
| 36 |
+
(~df['sz_top'].isna())]) > 0:
|
| 37 |
+
print('We found missing data')
|
| 38 |
+
df.loc[(~df['px'].isna())&
|
| 39 |
+
(df['in_zone'].isna())&
|
| 40 |
+
(~df['sz_top'].isna())&
|
| 41 |
+
(~df['pz'].isna())&
|
| 42 |
+
(~df['sz_bot'].isna())
|
| 43 |
+
,'in_zone'] = in_zone_model.predict(df.loc[(~df['px'].isna())&
|
| 44 |
+
(df['in_zone'].isna())&
|
| 45 |
+
(~df['sz_top'].isna())&
|
| 46 |
+
(~df['pz'].isna())&
|
| 47 |
+
(~df['sz_bot'].isna())][['px','pz','sz_top','sz_bot']].values)
|
| 48 |
+
hit_codes = ['single',
|
| 49 |
+
'double','home_run', 'triple']
|
| 50 |
+
|
| 51 |
+
ab_codes = ['single', 'strikeout', 'field_out',
|
| 52 |
+
'grounded_into_double_play', 'fielders_choice', 'force_out',
|
| 53 |
+
'double', 'field_error', 'home_run', 'triple',
|
| 54 |
+
'double_play',
|
| 55 |
+
'fielders_choice_out', 'strikeout_double_play',
|
| 56 |
+
'other_out','triple_play']
|
| 57 |
+
|
| 58 |
+
|
| 59 |
+
obp_true_codes = ['single', 'walk',
|
| 60 |
+
'double','home_run', 'triple',
|
| 61 |
+
'hit_by_pitch', 'intent_walk']
|
| 62 |
+
|
| 63 |
+
obp_codes = ['single', 'strikeout', 'walk', 'field_out',
|
| 64 |
+
'grounded_into_double_play', 'fielders_choice', 'force_out',
|
| 65 |
+
'double', 'sac_fly', 'field_error', 'home_run', 'triple',
|
| 66 |
+
'hit_by_pitch', 'double_play', 'intent_walk',
|
| 67 |
+
'fielders_choice_out', 'strikeout_double_play',
|
| 68 |
+
'sac_fly_double_play',
|
| 69 |
+
'other_out','triple_play']
|
| 70 |
+
|
| 71 |
+
|
| 72 |
+
contact_codes = ['In play, no out',
|
| 73 |
+
'Foul', 'In play, out(s)',
|
| 74 |
+
'In play, run(s)',
|
| 75 |
+
'Foul Bunt']
|
| 76 |
+
|
| 77 |
+
|
| 78 |
+
|
| 79 |
+
conditions_hit = [df.event_type.isin(hit_codes)]
|
| 80 |
+
choices_hit = [True]
|
| 81 |
+
df['hits'] = np.select(conditions_hit, choices_hit, default=False)
|
| 82 |
+
|
| 83 |
+
conditions_ab = [df.event_type.isin(ab_codes)]
|
| 84 |
+
choices_ab = [True]
|
| 85 |
+
df['ab'] = np.select(conditions_ab, choices_ab, default=False)
|
| 86 |
+
|
| 87 |
+
conditions_obp_true = [df.event_type.isin(obp_true_codes)]
|
| 88 |
+
choices_obp_true = [True]
|
| 89 |
+
df['on_base'] = np.select(conditions_obp_true, choices_obp_true, default=False)
|
| 90 |
+
|
| 91 |
+
conditions_obp = [df.event_type.isin(obp_codes)]
|
| 92 |
+
choices_obp = [True]
|
| 93 |
+
df['obp'] = np.select(conditions_obp, choices_obp, default=False)
|
| 94 |
+
|
| 95 |
+
bip_codes = ['In play, no out', 'In play, run(s)','In play, out(s)']
|
| 96 |
+
|
| 97 |
+
conditions_bip = [df.play_description.isin(bip_codes)]
|
| 98 |
+
choices_bip = [True]
|
| 99 |
+
df['bip'] = np.select(conditions_bip, choices_bip, default=False)
|
| 100 |
+
|
| 101 |
+
conditions = [
|
| 102 |
+
(df['launch_speed'].isna()),
|
| 103 |
+
(df['launch_speed']*1.5 - df['launch_angle'] >= 117 ) & (df['launch_speed'] + df['launch_angle'] >= 124) & (df['launch_speed'] > 98) & (df['launch_angle'] >= 8) & (df['launch_angle'] <= 50)
|
| 104 |
+
]
|
| 105 |
+
df['bip_div'] = ~df.launch_speed.isna()
|
| 106 |
+
choices = [False,True]
|
| 107 |
+
df['barrel'] = np.select(conditions, choices, default=np.nan)
|
| 108 |
+
df['barrel'] = loaded_model.predict(df[['launch_speed','launch_angle']].fillna(0).values)
|
| 109 |
+
conditions_ss = [
|
| 110 |
+
(df['launch_angle'].isna()),
|
| 111 |
+
(df['launch_angle'] >= 8 ) * (df['launch_angle'] <= 32 )
|
| 112 |
+
]
|
| 113 |
+
|
| 114 |
+
choices_ss = [False,True]
|
| 115 |
+
df['sweet_spot'] = np.select(conditions_ss, choices_ss, default=np.nan)
|
| 116 |
+
|
| 117 |
+
conditions_hh = [
|
| 118 |
+
(df['launch_speed'].isna()),
|
| 119 |
+
(df['launch_speed'] >= 94.5 )
|
| 120 |
+
]
|
| 121 |
+
|
| 122 |
+
choices_hh = [False,True]
|
| 123 |
+
df['hard_hit'] = np.select(conditions_hh, choices_hh, default=np.nan)
|
| 124 |
+
|
| 125 |
+
|
| 126 |
+
conditions_tb = [
|
| 127 |
+
(df['event_type']=='single'),
|
| 128 |
+
(df['event_type']=='double'),
|
| 129 |
+
(df['event_type']=='triple'),
|
| 130 |
+
(df['event_type']=='home_run'),
|
| 131 |
+
]
|
| 132 |
+
|
| 133 |
+
choices_tb = [1,2,3,4]
|
| 134 |
+
|
| 135 |
+
df['tb'] = np.select(conditions_tb, choices_tb, default=np.nan)
|
| 136 |
+
|
| 137 |
+
conditions_woba = [
|
| 138 |
+
(df['event_type'].isin(['strikeout', 'field_out', 'sac_fly', 'force_out',
|
| 139 |
+
'grounded_into_double_play', 'fielders_choice', 'field_error',
|
| 140 |
+
'sac_bunt', 'double_play', 'fielders_choice_out', 'strikeout_double_play',
|
| 141 |
+
'sac_fly_double_play', 'other_out'])),
|
| 142 |
+
(df['event_type']=='walk'),
|
| 143 |
+
(df['event_type']=='hit_by_pitch'),
|
| 144 |
+
(df['event_type']=='single'),
|
| 145 |
+
(df['event_type']=='double'),
|
| 146 |
+
(df['event_type']=='triple'),
|
| 147 |
+
(df['event_type']=='home_run'),
|
| 148 |
+
]
|
| 149 |
+
|
| 150 |
+
choices_woba = [0,
|
| 151 |
+
0.696,
|
| 152 |
+
0.726,
|
| 153 |
+
0.883,
|
| 154 |
+
1.244,
|
| 155 |
+
1.569,
|
| 156 |
+
2.004]
|
| 157 |
+
|
| 158 |
+
df['woba'] = np.select(conditions_woba, choices_woba, default=np.nan)
|
| 159 |
+
|
| 160 |
+
|
| 161 |
+
woba_codes = ['strikeout', 'field_out', 'single', 'walk', 'hit_by_pitch',
|
| 162 |
+
'double', 'sac_fly', 'force_out', 'home_run',
|
| 163 |
+
'grounded_into_double_play', 'fielders_choice', 'field_error',
|
| 164 |
+
'triple', 'sac_bunt', 'double_play',
|
| 165 |
+
'fielders_choice_out', 'strikeout_double_play',
|
| 166 |
+
'sac_fly_double_play', 'other_out']
|
| 167 |
+
|
| 168 |
+
|
| 169 |
+
|
| 170 |
+
|
| 171 |
+
|
| 172 |
+
|
| 173 |
+
conditions_woba_code = [
|
| 174 |
+
(df['event_type'].isin(woba_codes))
|
| 175 |
+
]
|
| 176 |
+
|
| 177 |
+
choices_woba_code = [1]
|
| 178 |
+
|
| 179 |
+
df['woba_codes'] = np.select(conditions_woba_code, choices_woba_code, default=np.nan)
|
| 180 |
+
|
| 181 |
+
|
| 182 |
+
df['woba_contact'] = [df['woba'].values[x] if df['bip'].values[x] == 1 else np.nan for x in range(len(df['woba_codes']))]
|
| 183 |
+
|
| 184 |
+
#df['in_zone'] = [x < 10 if type(x) == int else np.nan for x in df['zone']]
|
| 185 |
+
|
| 186 |
+
# df['in_zone_2'] = in_zone_model.predict(df[['x','y','sz_bot','sz_top']].fillna(0).values)
|
| 187 |
+
# df['in_zone_3'] = df['in_zone_2'] < 10
|
| 188 |
+
# df.loc[df['in_zone'].isna(),'in_zone'] = df.loc[df['in_zone'].isna(),'in_zone_3'].fillna(0)
|
| 189 |
+
|
| 190 |
+
|
| 191 |
+
df['whiffs'] = [1 if ((x == 'S')|(x == 'W')|(x =='T')) else 0 for x in df.play_code]
|
| 192 |
+
df['csw'] = [1 if ((x == 'S')|(x == 'W')|(x =='T')|(x == 'C')) else 0 for x in df.play_code]
|
| 193 |
+
df['swings'] = [1 if x == True else 0 for x in df.is_swing]
|
| 194 |
+
|
| 195 |
+
|
| 196 |
+
df['out_zone'] = df.in_zone == False
|
| 197 |
+
df['zone_swing'] = (df.in_zone == True)&(df.swings == 1)
|
| 198 |
+
df['zone_contact'] = (df.in_zone == True)&(df.swings == 1)&(df.whiffs == 0)
|
| 199 |
+
df['ozone_swing'] = (df.in_zone==False)&(df.swings == 1)
|
| 200 |
+
df['ozone_contact'] = (df.in_zone==False)&(df.swings == 1)&(df.whiffs == 0)
|
| 201 |
+
|
| 202 |
+
df['k'] = df.event_type.isin(list(filter(None, [x if 'strikeout' in x else '' for x in df.event_type.dropna().unique()])))
|
| 203 |
+
df['bb'] = df.event_type.isin(['walk','intent_walk'])
|
| 204 |
+
|
| 205 |
+
df['k_minus_bb'] = df['k'].astype(np.float32)-df['bb'].astype(np.float32)
|
| 206 |
+
df['bb_minus_k'] = df['bb'].astype(np.float32)-df['k'].astype(np.float32)
|
| 207 |
+
|
| 208 |
+
df['pa'] = [1 if isinstance(x, str) else 0 for x in df.event_type]
|
| 209 |
+
df['pitches'] = [1 if x else 0 for x in df.is_pitch]
|
| 210 |
+
|
| 211 |
+
|
| 212 |
+
df.loc[df['launch_speed'].isna(),'barrel'] = np.nan
|
| 213 |
+
|
| 214 |
+
|
| 215 |
+
pitch_cat = {'FA':'Fastball',
|
| 216 |
+
'FF':'Fastball',
|
| 217 |
+
'FT':'Fastball',
|
| 218 |
+
'FC':'Fastball',
|
| 219 |
+
'FS':'Off-Speed',
|
| 220 |
+
'FO':'Off-Speed',
|
| 221 |
+
'SI':'Fastball',
|
| 222 |
+
'ST':'Breaking',
|
| 223 |
+
'SL':'Breaking',
|
| 224 |
+
'CU':'Breaking',
|
| 225 |
+
'KC':'Breaking',
|
| 226 |
+
'SC':'Off-Speed',
|
| 227 |
+
'GY':'Off-Speed',
|
| 228 |
+
'SV':'Breaking',
|
| 229 |
+
'CS':'Breaking',
|
| 230 |
+
'CH':'Off-Speed',
|
| 231 |
+
'KN':'Off-Speed',
|
| 232 |
+
'EP':'Breaking',
|
| 233 |
+
'UN':np.nan,
|
| 234 |
+
'IN':np.nan,
|
| 235 |
+
'PO':np.nan,
|
| 236 |
+
'AB':np.nan,
|
| 237 |
+
'AS':np.nan,
|
| 238 |
+
'NP':np.nan}
|
| 239 |
+
#df['pitch_type'] = df['pitch_type'].map(pitch_cat).fillna('Unknown')
|
| 240 |
+
df['average'] = 'average'
|
| 241 |
+
|
| 242 |
+
df.loc[df['trajectory'] == 'bunt_popup','trajectory'] = 'popup'
|
| 243 |
+
df.loc[df['trajectory'] == 'bunt_grounder','trajectory'] = 'ground_ball'
|
| 244 |
+
df.loc[df['trajectory'] == '','trajectory'] = np.nan
|
| 245 |
+
df.loc[df['trajectory'] == 'bunt_line_drive','trajectory'] = 'line_drive'
|
| 246 |
+
df[['trajectory_fly_ball','trajectory_ground_ball','trajectory_line_drive','trajectory_popup']] = pd.get_dummies(df['trajectory'], prefix='trajectory')
|
| 247 |
+
|
| 248 |
+
df['attack_zone'] = np.nan
|
| 249 |
+
|
| 250 |
+
|
| 251 |
+
|
| 252 |
+
df.loc[df[['px','pz','sz_top','sz_bot']].isnull().sum(axis=1)==0,'attack_zone'] = attack_zone_model.predict(df.loc[df[['px','pz','sz_top','sz_bot']].isnull().sum(axis=1)==0][['px','pz','sz_top','sz_bot']])
|
| 253 |
+
|
| 254 |
+
|
| 255 |
+
|
| 256 |
+
df['heart'] = df['attack_zone'] == 0
|
| 257 |
+
df['shadow'] = df['attack_zone'] == 1
|
| 258 |
+
df['chase'] = df['attack_zone'] == 2
|
| 259 |
+
df['waste'] = df['attack_zone'] == 3
|
| 260 |
+
|
| 261 |
+
df['heart_swing'] = (df['attack_zone'] == 0)&(df['swings']==1)
|
| 262 |
+
df['shadow_swing'] = (df['attack_zone'] == 1)&(df['swings']==1)
|
| 263 |
+
df['chase_swing'] = (df['attack_zone'] == 2)&(df['swings']==1)
|
| 264 |
+
df['waste_swing'] = (df['attack_zone'] == 3)&(df['swings']==1)
|
| 265 |
+
|
| 266 |
+
df['heart_whiff'] = (df['attack_zone'] == 0)&(df['whiffs']==1)
|
| 267 |
+
df['shadow_whiff'] = (df['attack_zone'] == 1)&(df['whiffs']==1)
|
| 268 |
+
df['chase_whiff'] = (df['attack_zone'] == 2)&(df['whiffs']==1)
|
| 269 |
+
df['waste_whiff'] = (df['attack_zone'] == 3)&(df['whiffs']==1)
|
| 270 |
+
|
| 271 |
+
df['woba_pred'] = np.nan
|
| 272 |
+
df['woba_pred_contact'] = np.nan
|
| 273 |
+
|
| 274 |
+
if len(df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'woba_pred']) > 0:
|
| 275 |
+
|
| 276 |
+
|
| 277 |
+
df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'woba_pred'] = [sum(x) for x in xwoba_model.predict_proba(df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0][['launch_angle','launch_speed']]) * ([0, 0.883,1.244,1.569,2.004])]
|
| 278 |
+
|
| 279 |
+
## Assign a value of 0.696 to every walk in the dataset
|
| 280 |
+
df.loc[df['event_type'].isin(['walk']),'woba_pred'] = 0.696
|
| 281 |
+
|
| 282 |
+
## Assign a value of 0.726 to every hit by pitch in the dataset
|
| 283 |
+
df.loc[df['event_type'].isin(['hit_by_pitch']),'woba_pred'] = 0.726
|
| 284 |
+
|
| 285 |
+
## Assign a value of 0 to every Strikeout in the dataset
|
| 286 |
+
df.loc[df['event_type'].isin(['strikeout','strikeout_double_play']),'woba_pred'] = 0
|
| 287 |
+
|
| 288 |
+
|
| 289 |
+
df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'woba_pred_contact'] = [sum(x) for x in xwoba_model.predict_proba(df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0][['launch_angle','launch_speed']]) * ([0, 0.883,1.244,1.569,2.004])]
|
| 290 |
+
|
| 291 |
+
df['xwoba_codes'] = np.nan
|
| 292 |
+
df.loc[df[['launch_angle','launch_speed']].isnull().sum(axis=1)==0,'xwoba_codes'] = 1
|
| 293 |
+
## Assign a value of 0.696 to every walk in the dataset
|
| 294 |
+
df.loc[df['event_type'].isin(['walk']),'xwoba_codes'] = 1
|
| 295 |
+
|
| 296 |
+
## Assign a value of 0.726 to every hit by pitch in the dataset
|
| 297 |
+
df.loc[df['event_type'].isin(['hit_by_pitch']),'xwoba_codes'] = 1
|
| 298 |
+
|
| 299 |
+
## Assign a value of 0 to every Strikeout in the dataset
|
| 300 |
+
df.loc[df['event_type'].isin(['strikeout','strikeout_double_play']),'xwoba_codes'] = 1
|
| 301 |
+
return df
|
| 302 |
+
|
| 303 |
+
def df_update_summ(df=pd.DataFrame()):
|
| 304 |
+
df_summ = df.groupby(['pitcher_id','pitcher_name']).agg(
|
| 305 |
+
pa = ('pa','sum'),
|
| 306 |
+
ab = ('ab','sum'),
|
| 307 |
+
obp_pa = ('obp','sum'),
|
| 308 |
+
hits = ('hits','sum'),
|
| 309 |
+
on_base = ('on_base','sum'),
|
| 310 |
+
k = ('k','sum'),
|
| 311 |
+
bb = ('bb','sum'),
|
| 312 |
+
bb_minus_k = ('bb_minus_k','sum'),
|
| 313 |
+
csw = ('csw','sum'),
|
| 314 |
+
bip = ('bip','sum'),
|
| 315 |
+
bip_div = ('bip_div','sum'),
|
| 316 |
+
tb = ('tb','sum'),
|
| 317 |
+
woba = ('woba','sum'),
|
| 318 |
+
woba_contact = ('woba_contact','sum'),
|
| 319 |
+
xwoba = ('woba_pred','sum'),
|
| 320 |
+
xwoba_contact = ('woba_pred_contact','sum'),
|
| 321 |
+
woba_codes = ('woba_codes','sum'),
|
| 322 |
+
xwoba_codes = ('xwoba_codes','sum'),
|
| 323 |
+
hard_hit = ('hard_hit','sum'),
|
| 324 |
+
barrel = ('barrel','sum'),
|
| 325 |
+
sweet_spot = ('sweet_spot','sum'),
|
| 326 |
+
max_launch_speed = ('launch_speed','max'),
|
| 327 |
+
launch_speed_90 = ('launch_speed',percentile(90)),
|
| 328 |
+
launch_speed = ('launch_speed','mean'),
|
| 329 |
+
launch_angle = ('launch_angle','mean'),
|
| 330 |
+
pitches = ('is_pitch','sum'),
|
| 331 |
+
swings = ('swings','sum'),
|
| 332 |
+
in_zone = ('in_zone','sum'),
|
| 333 |
+
out_zone = ('out_zone','sum'),
|
| 334 |
+
whiffs = ('whiffs','sum'),
|
| 335 |
+
zone_swing = ('zone_swing','sum'),
|
| 336 |
+
zone_contact = ('zone_contact','sum'),
|
| 337 |
+
ozone_swing = ('ozone_swing','sum'),
|
| 338 |
+
ozone_contact = ('ozone_contact','sum'),
|
| 339 |
+
ground_ball = ('trajectory_ground_ball','sum'),
|
| 340 |
+
line_drive = ('trajectory_line_drive','sum'),
|
| 341 |
+
fly_ball =('trajectory_fly_ball','sum'),
|
| 342 |
+
pop_up = ('trajectory_popup','sum'),
|
| 343 |
+
attack_zone = ('attack_zone','count'),
|
| 344 |
+
heart = ('heart','sum'),
|
| 345 |
+
shadow = ('shadow','sum'),
|
| 346 |
+
chase = ('chase','sum'),
|
| 347 |
+
waste = ('waste','sum'),
|
| 348 |
+
heart_swing = ('heart_swing','sum'),
|
| 349 |
+
shadow_swing = ('shadow_swing','sum'),
|
| 350 |
+
chase_swing = ('chase_swing','sum'),
|
| 351 |
+
waste_swing = ('waste_swing','sum'),
|
| 352 |
+
).reset_index()
|
| 353 |
+
return df_summ
|
| 354 |
+
|
| 355 |
+
def df_update_summ_avg(df=pd.DataFrame()):
|
| 356 |
+
df_summ_avg = df.groupby(['average']).agg(
|
| 357 |
+
|
| 358 |
+
).reset_index()
|
| 359 |
+
return df_summ_avg
|
| 360 |
+
|
| 361 |
+
def df_summ_changes(df_summ=pd.DataFrame()):
|
| 362 |
+
df_summ['avg'] = [df_summ.hits[x]/df_summ.ab[x] if df_summ.ab[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 363 |
+
df_summ['obp'] = [df_summ.on_base[x]/df_summ.obp_pa[x] if df_summ.obp_pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 364 |
+
df_summ['slg'] = [df_summ.tb[x]/df_summ.ab[x] if df_summ.ab[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 365 |
+
|
| 366 |
+
df_summ['ops'] = df_summ['obp']+df_summ['slg']
|
| 367 |
+
|
| 368 |
+
df_summ['k_percent'] = [df_summ.k[x]/df_summ.pa[x] if df_summ.pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 369 |
+
df_summ['bb_percent'] =[df_summ.bb[x]/df_summ.pa[x] if df_summ.pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 370 |
+
df_summ['bb_minus_k_percent'] =[(df_summ.bb_minus_k[x])/df_summ.pa[x] if df_summ.pa[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 371 |
+
|
| 372 |
+
df_summ['bb_over_k_percent'] =[df_summ.bb[x]/df_summ.k[x] if df_summ.k[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 373 |
+
|
| 374 |
+
|
| 375 |
+
|
| 376 |
+
|
| 377 |
+
df_summ['csw_percent'] =[df_summ.csw[x]/df_summ.pitches[x] if df_summ.pitches[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 378 |
+
|
| 379 |
+
|
| 380 |
+
df_summ['sweet_spot_percent'] = [df_summ.sweet_spot[x]/df_summ.bip_div[x] if df_summ.bip_div[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 381 |
+
|
| 382 |
+
df_summ['woba_percent'] = [df_summ.woba[x]/df_summ.woba_codes[x] if df_summ.woba_codes[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 383 |
+
df_summ['woba_percent_contact'] = [df_summ.woba_contact[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 384 |
+
#df_summ['hard_hit_percent'] = [df_summ.sweet_spot[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 385 |
+
df_summ['hard_hit_percent'] = [df_summ.hard_hit[x]/df_summ.bip_div[x] if df_summ.bip_div[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 386 |
+
|
| 387 |
+
|
| 388 |
+
df_summ['barrel_percent'] = [df_summ.barrel[x]/df_summ.bip_div[x] if df_summ.bip_div[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 389 |
+
|
| 390 |
+
df_summ['zone_contact_percent'] = [df_summ.zone_contact[x]/df_summ.zone_swing[x] if df_summ.zone_swing[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 391 |
+
|
| 392 |
+
df_summ['zone_swing_percent'] = [df_summ.zone_swing[x]/df_summ.in_zone[x] if df_summ.in_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 393 |
+
|
| 394 |
+
df_summ['zone_percent'] = [df_summ.in_zone[x]/df_summ.pitches[x] if df_summ.pitches[x] > 0 else np.nan for x in range(len(df_summ))]
|
| 395 |
+
|
| 396 |
+
df_summ['chase_percent'] = [df_summ.ozone_swing[x]/(df_summ.pitches[x] - df_summ.in_zone[x]) if (df_summ.pitches[x]- df_summ.in_zone[x]) != 0 else np.nan for x in range(len(df_summ))]
|
| 397 |
+
|
| 398 |
+
df_summ['chase_contact'] = [df_summ.ozone_contact[x]/df_summ.ozone_swing[x] if df_summ.ozone_swing[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 399 |
+
|
| 400 |
+
df_summ['swing_percent'] = [df_summ.swings[x]/df_summ.pitches[x] if df_summ.pitches[x] > 0 else np.nan for x in range(len(df_summ))]
|
| 401 |
+
|
| 402 |
+
df_summ['whiff_rate'] = [df_summ.whiffs[x]/df_summ.swings[x] if df_summ.swings[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 403 |
+
|
| 404 |
+
df_summ['swstr_rate'] = [df_summ.whiffs[x]/df_summ.pitches[x] if df_summ.pitches[x] > 0 else np.nan for x in range(len(df_summ))]
|
| 405 |
+
|
| 406 |
+
df_summ['ground_ball_percent'] = [df_summ.ground_ball[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 407 |
+
|
| 408 |
+
df_summ['line_drive_percent'] = [df_summ.line_drive[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 409 |
+
|
| 410 |
+
df_summ['fly_ball_percent'] = [df_summ.fly_ball[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 411 |
+
|
| 412 |
+
df_summ['pop_up_percent'] = [df_summ.pop_up[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 413 |
+
|
| 414 |
+
|
| 415 |
+
|
| 416 |
+
df_summ['heart_zone_percent'] = [df_summ.heart[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 417 |
+
|
| 418 |
+
df_summ['shadow_zone_percent'] = [df_summ.shadow[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 419 |
+
|
| 420 |
+
df_summ['chase_zone_percent'] = [df_summ.chase[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 421 |
+
|
| 422 |
+
df_summ['waste_zone_percent'] = [df_summ.waste[x]/df_summ.attack_zone[x] if df_summ.attack_zone[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 423 |
+
|
| 424 |
+
|
| 425 |
+
df_summ['heart_zone_swing_percent'] = [df_summ.heart_swing[x]/df_summ.heart[x] if df_summ.heart[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 426 |
+
|
| 427 |
+
df_summ['shadow_zone_swing_percent'] = [df_summ.shadow_swing[x]/df_summ.shadow[x] if df_summ.shadow[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 428 |
+
|
| 429 |
+
df_summ['chase_zone_swing_percent'] = [df_summ.chase_swing[x]/df_summ.chase[x] if df_summ.chase[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 430 |
+
|
| 431 |
+
df_summ['waste_zone_swing_percent'] = [df_summ.waste_swing[x]/df_summ.waste[x] if df_summ.waste[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 432 |
+
|
| 433 |
+
|
| 434 |
+
|
| 435 |
+
|
| 436 |
+
df_summ['xwoba_percent'] = [df_summ.xwoba[x]/df_summ.xwoba_codes[x] if df_summ.xwoba_codes[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 437 |
+
df_summ['xwoba_percent_contact'] = [df_summ.xwoba_contact[x]/df_summ.bip[x] if df_summ.bip[x] != 0 else np.nan for x in range(len(df_summ))]
|
| 438 |
+
|
| 439 |
+
df_summ = df_summ.dropna(subset=['bip'])
|
| 440 |
+
return df_summ
|
| 441 |
+
|
| 442 |
+
def df_summ_filter_out(df_summ=pd.DataFrame(),batter_select = 0):
|
| 443 |
+
df_summ_filter = df_summ[df_summ['pa'] >= min(math.floor(df_summ.xs(batter_select,level=0)['pa']/10)*10,500)]
|
| 444 |
+
df_summ_filter_pct = df_summ_filter.rank(pct=True,ascending=True)
|
| 445 |
+
df_summ_player = df_summ.xs(batter_select,level=0)
|
| 446 |
+
df_summ_player_pct = df_summ_filter_pct.xs(batter_select,level=0)
|
| 447 |
+
return df_summ_filter,df_summ_filter_pct,df_summ_player,df_summ_player_pct
|
| 448 |
+
|
| 449 |
+
def df_summ_batter_pitch_up(df=pd.DataFrame()):
|
| 450 |
+
df_summ_batter_pitch = df.dropna(subset=['pitch_type']).groupby(['pitcher_id','pitcher_name','pitch_type']).agg(
|
| 451 |
+
pa = ('pa','sum'),
|
| 452 |
+
ab = ('ab','sum'),
|
| 453 |
+
obp_pa = ('obp','sum'),
|
| 454 |
+
hits = ('hits','sum'),
|
| 455 |
+
on_base = ('on_base','sum'),
|
| 456 |
+
k = ('k','sum'),
|
| 457 |
+
bb = ('bb','sum'),
|
| 458 |
+
bb_minus_k = ('bb_minus_k','sum'),
|
| 459 |
+
csw = ('csw','sum'),
|
| 460 |
+
bip = ('bip','sum'),
|
| 461 |
+
bip_div = ('bip_div','sum'),
|
| 462 |
+
tb = ('tb','sum'),
|
| 463 |
+
woba = ('woba','sum'),
|
| 464 |
+
woba_contact = ('woba_pred_contact','sum'),
|
| 465 |
+
xwoba = ('woba_pred','sum'),
|
| 466 |
+
xwoba_contact = ('woba_pred','sum'),
|
| 467 |
+
woba_codes = ('woba_codes','sum'),
|
| 468 |
+
xwoba_codes = ('xwoba_codes','sum'),
|
| 469 |
+
hard_hit = ('hard_hit','sum'),
|
| 470 |
+
barrel = ('barrel','sum'),
|
| 471 |
+
sweet_spot = ('sweet_spot','sum'),
|
| 472 |
+
max_launch_speed = ('launch_speed','max'),
|
| 473 |
+
launch_speed_90 = ('launch_speed',percentile(90)),
|
| 474 |
+
launch_speed = ('launch_speed','mean'),
|
| 475 |
+
launch_angle = ('launch_angle','mean'),
|
| 476 |
+
pitches = ('is_pitch','sum'),
|
| 477 |
+
swings = ('swings','sum'),
|
| 478 |
+
in_zone = ('in_zone','sum'),
|
| 479 |
+
out_zone = ('out_zone','sum'),
|
| 480 |
+
whiffs = ('whiffs','sum'),
|
| 481 |
+
zone_swing = ('zone_swing','sum'),
|
| 482 |
+
zone_contact = ('zone_contact','sum'),
|
| 483 |
+
ozone_swing = ('ozone_swing','sum'),
|
| 484 |
+
ozone_contact = ('ozone_contact','sum'),
|
| 485 |
+
ground_ball = ('trajectory_ground_ball','sum'),
|
| 486 |
+
line_drive = ('trajectory_line_drive','sum'),
|
| 487 |
+
fly_ball =('trajectory_fly_ball','sum'),
|
| 488 |
+
pop_up = ('trajectory_popup','sum'),
|
| 489 |
+
attack_zone = ('attack_zone','count'),
|
| 490 |
+
heart = ('heart','sum'),
|
| 491 |
+
shadow = ('shadow','sum'),
|
| 492 |
+
chase = ('chase','sum'),
|
| 493 |
+
waste = ('waste','sum'),
|
| 494 |
+
heart_swing = ('heart_swing','sum'),
|
| 495 |
+
shadow_swing = ('shadow_swing','sum'),
|
| 496 |
+
chase_swing = ('chase_swing','sum'),
|
| 497 |
+
waste_swing = ('waste_swing','sum'),
|
| 498 |
+
).reset_index()
|
| 499 |
+
|
| 500 |
+
#return df_summ_batter_pitch
|
| 501 |
+
df_summ_batter_pitch['avg'] = [df_summ_batter_pitch.hits[x]/df_summ_batter_pitch.ab[x] if df_summ_batter_pitch.ab[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 502 |
+
df_summ_batter_pitch['obp'] = [df_summ_batter_pitch.on_base[x]/df_summ_batter_pitch.obp_pa[x] if df_summ_batter_pitch.obp_pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 503 |
+
df_summ_batter_pitch['slg'] = [df_summ_batter_pitch.tb[x]/df_summ_batter_pitch.ab[x] if df_summ_batter_pitch.ab[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 504 |
+
|
| 505 |
+
df_summ_batter_pitch['ops'] = df_summ_batter_pitch['obp']+df_summ_batter_pitch['slg']
|
| 506 |
+
|
| 507 |
+
df_summ_batter_pitch['k_percent'] = [df_summ_batter_pitch.k[x]/df_summ_batter_pitch.pa[x] if df_summ_batter_pitch.pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 508 |
+
df_summ_batter_pitch['bb_percent'] =[df_summ_batter_pitch.bb[x]/df_summ_batter_pitch.pa[x] if df_summ_batter_pitch.pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 509 |
+
df_summ_batter_pitch['bb_minus_k_percent'] =[(df_summ_batter_pitch.bb_minus_k[x])/df_summ_batter_pitch.pa[x] if df_summ_batter_pitch.pa[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 510 |
+
|
| 511 |
+
df_summ_batter_pitch['bb_over_k_percent'] =[df_summ_batter_pitch.bb[x]/df_summ_batter_pitch.k[x] if df_summ_batter_pitch.k[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 512 |
+
|
| 513 |
+
|
| 514 |
+
|
| 515 |
+
|
| 516 |
+
df_summ_batter_pitch['csw_percent'] =[df_summ_batter_pitch.csw[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 517 |
+
|
| 518 |
+
|
| 519 |
+
df_summ_batter_pitch['sweet_spot_percent'] = [df_summ_batter_pitch.sweet_spot[x]/df_summ_batter_pitch.bip_div[x] if df_summ_batter_pitch.bip_div[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 520 |
+
|
| 521 |
+
df_summ_batter_pitch['woba_percent'] = [df_summ_batter_pitch.woba[x]/df_summ_batter_pitch.woba_codes[x] if df_summ_batter_pitch.woba_codes[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 522 |
+
df_summ_batter_pitch['woba_percent_contact'] = [df_summ_batter_pitch.woba_contact[x]/df_summ_batter_pitch.bip[x] if df_summ_batter_pitch.bip[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 523 |
+
#df_summ_batter_pitch['hard_hit_percent'] = [df_summ_batter_pitch.sweet_spot[x]/df_summ_batter_pitch.bip[x] if df_summ_batter_pitch.bip[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 524 |
+
df_summ_batter_pitch['hard_hit_percent'] = [df_summ_batter_pitch.hard_hit[x]/df_summ_batter_pitch.bip_div[x] if df_summ_batter_pitch.bip_div[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 525 |
+
|
| 526 |
+
|
| 527 |
+
df_summ_batter_pitch['barrel_percent'] = [df_summ_batter_pitch.barrel[x]/df_summ_batter_pitch.bip_div[x] if df_summ_batter_pitch.bip_div[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 528 |
+
|
| 529 |
+
df_summ_batter_pitch['zone_contact_percent'] = [df_summ_batter_pitch.zone_contact[x]/df_summ_batter_pitch.zone_swing[x] if df_summ_batter_pitch.zone_swing[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 530 |
+
|
| 531 |
+
df_summ_batter_pitch['zone_swing_percent'] = [df_summ_batter_pitch.zone_swing[x]/df_summ_batter_pitch.in_zone[x] if df_summ_batter_pitch.in_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 532 |
+
|
| 533 |
+
df_summ_batter_pitch['zone_percent'] = [df_summ_batter_pitch.in_zone[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 534 |
+
|
| 535 |
+
df_summ_batter_pitch['chase_percent'] = [df_summ_batter_pitch.ozone_swing[x]/(df_summ_batter_pitch.pitches[x] - df_summ_batter_pitch.in_zone[x]) if (df_summ_batter_pitch.pitches[x]- df_summ_batter_pitch.in_zone[x]) != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 536 |
+
|
| 537 |
+
df_summ_batter_pitch['chase_contact'] = [df_summ_batter_pitch.ozone_contact[x]/df_summ_batter_pitch.ozone_swing[x] if df_summ_batter_pitch.ozone_swing[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 538 |
+
|
| 539 |
+
df_summ_batter_pitch['swing_percent'] = [df_summ_batter_pitch.swings[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 540 |
+
|
| 541 |
+
df_summ_batter_pitch['whiff_rate'] = [df_summ_batter_pitch.whiffs[x]/df_summ_batter_pitch.swings[x] if df_summ_batter_pitch.swings[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 542 |
+
|
| 543 |
+
df_summ_batter_pitch['swstr_rate'] = [df_summ_batter_pitch.whiffs[x]/df_summ_batter_pitch.pitches[x] if df_summ_batter_pitch.pitches[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 544 |
+
|
| 545 |
+
df_summ_batter_pitch['heart_zone_percent'] = [df_summ_batter_pitch.heart[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 546 |
+
|
| 547 |
+
df_summ_batter_pitch['shadow_zone_percent'] = [df_summ_batter_pitch.shadow[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 548 |
+
|
| 549 |
+
df_summ_batter_pitch['chase_zone_percent'] = [df_summ_batter_pitch.chase[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 550 |
+
|
| 551 |
+
df_summ_batter_pitch['waste_zone_percent'] = [df_summ_batter_pitch.waste[x]/df_summ_batter_pitch.attack_zone[x] if df_summ_batter_pitch.attack_zone[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 552 |
+
|
| 553 |
+
|
| 554 |
+
df_summ_batter_pitch['heart_zone_swing_percent'] = [df_summ_batter_pitch.heart_swing[x]/df_summ_batter_pitch.heart[x] if df_summ_batter_pitch.heart[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 555 |
+
|
| 556 |
+
df_summ_batter_pitch['shadow_zone_swing_percent'] = [df_summ_batter_pitch.shadow_swing[x]/df_summ_batter_pitch.shadow[x] if df_summ_batter_pitch.shadow[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 557 |
+
|
| 558 |
+
df_summ_batter_pitch['chase_zone_swing_percent'] = [df_summ_batter_pitch.chase_swing[x]/df_summ_batter_pitch.chase[x] if df_summ_batter_pitch.chase[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 559 |
+
|
| 560 |
+
df_summ_batter_pitch['waste_zone_swing_percent'] = [df_summ_batter_pitch.waste_swing[x]/df_summ_batter_pitch.waste[x] if df_summ_batter_pitch.waste[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 561 |
+
|
| 562 |
+
|
| 563 |
+
|
| 564 |
+
|
| 565 |
+
df_summ_batter_pitch['xwoba_percent'] = [df_summ_batter_pitch.xwoba[x]/df_summ_batter_pitch.xwoba_codes[x] if df_summ_batter_pitch.xwoba_codes[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 566 |
+
df_summ_batter_pitch['xwoba_percent_contact'] = [df_summ_batter_pitch.xwoba_contact[x]/df_summ_batter_pitch.bip[x] if df_summ_batter_pitch.bip[x] != 0 else np.nan for x in range(len(df_summ_batter_pitch))]
|
| 567 |
+
|
| 568 |
+
|
| 569 |
+
|
| 570 |
+
|
| 571 |
+
df_summ_batter_pitch['bip'] = df_summ_batter_pitch['bip'].fillna(0)
|
| 572 |
+
|
| 573 |
+
return df_summ_batter_pitch
|