File size: 2,000 Bytes
d6ea71e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
"""Utility functions for working with SPADL dataframes."""
from typing import cast
from pandera.typing import DataFrame
from . import config as spadlconfig
from .schema import SPADLSchema
def add_names(actions: DataFrame[SPADLSchema]) -> DataFrame[SPADLSchema]:
"""Add the type name, result name and bodypart name to a SPADL dataframe.
Parameters
----------
actions : pd.DataFrame
A SPADL dataframe.
Returns
-------
pd.DataFrame
The original dataframe with a 'type_name', 'result_name' and
'bodypart_name' appended.
"""
return cast(
DataFrame[SPADLSchema],
actions.drop(columns=["type_name", "result_name", "bodypart_name"], errors="ignore")
.merge(spadlconfig.actiontypes_df(), how="left")
.merge(spadlconfig.results_df(), how="left")
.merge(spadlconfig.bodyparts_df(), how="left")
.set_index(actions.index),
)
def play_left_to_right(
actions: DataFrame[SPADLSchema], home_team_id: int
) -> DataFrame[SPADLSchema]:
"""Perform all action in the same playing direction.
This changes the start and end location of each action, such that all actions
are performed as if the team that executes the action plays from left to
right.
Parameters
----------
actions : pd.DataFrame
The SPADL actins of a game.
home_team_id : int
The ID of the home team.
Returns
-------
list(pd.DataFrame)
All actions performed left to right.
See Also
--------
socceraction.vaep.features.play_left_to_right : For transforming gamestates.
"""
ltr_actions = actions.copy()
away_idx = actions.team_id != home_team_id
for col in ["start_x", "end_x"]:
ltr_actions.loc[away_idx, col] = spadlconfig.field_length - actions[away_idx][col].values
for col in ["start_y", "end_y"]:
ltr_actions.loc[away_idx, col] = spadlconfig.field_width - actions[away_idx][col].values
return ltr_actions
|