Merge branch 'main' of https://github.com/sportsdataverse/sportsdatav…

…erse-py

sportsdataverse/cfb/cfb_pbp.py

@@ -5057,6 +5057,16 @@ def weighted_mean(s, df, wcol):
             EPA_success_rate = ('EPA_success', mean),
         )
 
+        situation_box_rz = self.plays_json[(self.plays_json.rz_play == True)].groupby(by=["pos_team"], as_index=False).agg(
+            EPA_success_rz = ('EPA_success', sum),
+            EPA_success_rate_rz = ('EPA_success', mean),
+        )
+
+        situation_box_third = self.plays_json[(self.plays_json.down == 3)].groupby(by=["pos_team"], as_index=False).agg(
+            EPA_success_third = ('EPA_success', sum),
+            EPA_success_rate_third = ('EPA_success', mean),
+        )
+
         situation_box_pass = self.plays_json[(self.plays_json["pass"] == True) & (self.plays_json.scrimmage_play == True)].groupby(by=["pos_team"], as_index=False).agg(
             EPA_success_pass = ('EPA_success', sum),
             EPA_success_pass_rate = ('EPA_success', mean),
@@ -5151,7 +5161,7 @@ def weighted_mean(s, df, wcol):
             EPA_passing_down = ('EPA_success_standard_down', sum),
             EPA_passing_down_per_play = ('EPA_success_standard_down', mean)
         )
-        situation_data_frames = [situation_box_normal, situation_box_pass, situation_box_rush, situation_box_early, situation_box_early_pass, situation_box_early_rush, situation_box_middle8, situation_box_middle8_pass, situation_box_middle8_rush, situation_box_late, situation_box_standard, situation_box_passing]
+        situation_data_frames = [situation_box_normal, situation_box_pass, situation_box_rush, situation_box_rz, situation_box_third, situation_box_early, situation_box_early_pass, situation_box_early_rush, situation_box_middle8, situation_box_middle8_pass, situation_box_middle8_rush, situation_box_late, situation_box_standard, situation_box_passing]
         situation_box = reduce(lambda left,right: pd.merge(left,right,on=['pos_team'], how='outer'), situation_data_frames)
         situation_box = situation_box.replace({np.nan:None})