Update functions/PitchPlotFunctions.py

functions/PitchPlotFunctions.py

@@ -14,6 +14,8 @@ import matplotlib.patches as mpatches
 import matplotlib.lines as mlines
 from matplotlib.figure import Figure
 import api_scraper
+from stuff_model import calculate_arm_angles as caa
+
 
 # Initialize the scraper
 scraper = api_scraper.MLB_Scrape()
@@ -381,6 +383,17 @@ class PitchPlotFunctions:
         # Set plot limits and labels
         ax.set_xlim((-25, 25))
         ax.set_ylim((-25, 25))
+
+        df_aa = caa.calculate_arm_angles(df,df['pitcher_id'][0])['arm_angle']
+    
+        # Plot average arm angle
+        mean_arm_angle = df_aa.mean()
+        x_end = 30
+        y_end = x_end * np.tan(np.radians(mean_arm_angle))
+        ax.plot([0, x_end], [0, y_end], color='grey', linestyle='--', linewidth=2,zorder=0)
+    
+
+        
         ax.hlines(y=0, xmin=-50, xmax=50, color=colour_palette[8], alpha=0.5, linestyles='--', zorder=1)
         ax.vlines(x=0, ymin=-50, ymax=50, color=colour_palette[8], alpha=0.5, linestyles='--', zorder=1)
         ax.set_xlabel('Horizontal Break (in)', fontdict=font_properties_axes)