def transform(self, *args, **kwargs):
"""
See [transform][kloppy.helpers.transform]
"""
from kloppy import transform
return transform(self, *args, **kwargs)
def main():
"""
This example shows the use of Statsbomb datasets, and how we can pass argument
to the dataset loader.
"""
logging.basicConfig(
stream=sys.stdout,
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
logger = logging.getLogger(__name__)
dataset = datasets.load(
"statsbomb", {"event_types": ["pass", "take_on", "carry", "shot"]})
with performance_logging("transform", logger=logger):
# convert to TRACAB coordinates
dataset = transform(dataset,
to_orientation="FIXED_HOME_AWAY",
to_pitch_dimensions=[(-5500, 5500), (-3300, 3300)])
with performance_logging("to pandas", logger=logger):
dataframe = to_pandas(dataset)
print(dataframe[:50].to_string())
def test_transform(self):
tracking_data = self._get_dataset()
# orientation change AND dimension scale
transformed_data_set = transform(tracking_data,
to_orientation="AWAY_TEAM",
to_pitch_dimensions=[[0, 1], [0, 1]])
assert transformed_data_set.frames[0].ball_position == Point(x=0, y=1)
assert transformed_data_set.frames[1].ball_position == Point(x=1, y=0)
def main():
logging.basicConfig(
stream=sys.stdout,
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
logger = logging.getLogger(__name__)
dataset = datasets.load(
"statsbomb", {"event_types": ["pass", "take_on", "carry", "shot"]})
with performance_logging("transform", logger=logger):
dataset = transform(dataset,
to_orientation="FIXED_HOME_AWAY",
to_pitch_dimensions=[(-5500, 5500), (-3300, 3300)])
with performance_logging("to pandas", logger=logger):
dataframe = to_pandas(dataset)
print(dataframe[:50].to_string())
def transform(self, *args, **kwargs):
from kloppy import transform
return transform(self, *args, **kwargs)
