Upload calculate_arm_angles.py

stuff_model/calculate_arm_angles.py

@@ -5,7 +5,7 @@ import requests
 def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
     df_arm_angle = pl.read_csv('stuff_model/pitcher_arm_angles_2024.csv')
     #pitcher_id = 489446
-    df_filter = df.filter(pl.col("pitcher_id") == pitcher_id)
+    df_filter = df.filter(pl.col("pitcher_id") == pitcher_id).drop_nulls(subset=["release_pos_x", "release_pos_z"])
     # data = requests.get(f'https://statsapi.mlb.com/api/v1/people?personIds={pitcher_id}').json()
 
     if pitcher_id not in df_arm_angle["pitcher"]:
@@ -45,7 +45,7 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
             )
             .with_columns(
                 (pl.col("release_pos_z") - shoulder_z).alias("Opp"),
-                (pl.col("release_pos_x") - shoulder_x).alias("Adj"),
+                (pl.col("release_pos_x") - shoulder_x*-1).alias("Adj"),
             )
             .with_columns(
                 pl.struct(["Opp", "Adj"]).map_elements(lambda x: np.arctan2(x["Opp"], x["Adj"])).alias("arm_angle_rad")
@@ -57,11 +57,10 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
             )
     
         df_filter = df_filter.with_columns(
-                ((pl.col("arm_angle") * 0.25) + (ball_angle * 0.75)).alias("arm_angle")
+                ((pl.col("arm_angle") * 1) + (ball_angle * 0.0)).alias("arm_angle")
             )
     
 
 
     
-    return df_filter
-
+    return df_filter