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)