File size: 2,171 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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
"""Configuration of the SPADL language.
Attributes
----------
field_length : float
The length of a pitch (in meters).
field_width : float
The width of a pitch (in meters).
bodyparts : list(str)
The bodyparts used in the SPADL language.
results : list(str)
The action results used in the SPADL language.
actiontypes : list(str)
The action types used in the SPADL language.
"""
import pandas as pd # type: ignore
field_length: float = 105.0 # unit: meters
field_width: float = 68.0 # unit: meters
bodyparts: list[str] = ["foot", "head", "other", "head/other", "foot_left", "foot_right"]
results: list[str] = [
"fail",
"success",
"offside",
"owngoal",
"yellow_card",
"red_card",
]
actiontypes: list[str] = [
"pass",
"cross",
"throw_in",
"freekick_crossed",
"freekick_short",
"corner_crossed",
"corner_short",
"take_on",
"foul",
"tackle",
"interception",
"shot",
"shot_penalty",
"shot_freekick",
"keeper_save",
"keeper_claim",
"keeper_punch",
"keeper_pick_up",
"clearance",
"bad_touch",
"non_action",
"dribble",
"goalkick",
]
def actiontypes_df() -> pd.DataFrame:
"""Return a dataframe with the type id and type name of each SPADL action type.
Returns
-------
pd.DataFrame
The 'type_id' and 'type_name' of each SPADL action type.
"""
return pd.DataFrame(list(enumerate(actiontypes)), columns=["type_id", "type_name"])
def results_df() -> pd.DataFrame:
"""Return a dataframe with the result id and result name of each SPADL action type.
Returns
-------
pd.DataFrame
The 'result_id' and 'result_name' of each SPADL action type.
"""
return pd.DataFrame(list(enumerate(results)), columns=["result_id", "result_name"])
def bodyparts_df() -> pd.DataFrame:
"""Return a dataframe with the bodypart id and bodypart name of each SPADL action type.
Returns
-------
pd.DataFrame
The 'bodypart_id' and 'bodypart_name' of each SPADL action type.
"""
return pd.DataFrame(list(enumerate(bodyparts)), columns=["bodypart_id", "bodypart_name"])
|