Added clean_locations helper function

pystatsbomb/helpers.py

@@ -1 +1,45 @@
-raise NotImplementedError("To be released in a future version")
+import numpy as np
+
+
+def all_clean(df):
+    df = clean_locations(df)
+    df = goalkeeper_info(df)
+    df = shot_info(df)
+    df = freeze_frame_info(df)
+    df = format_elapsed_time(df)
+    df = possession_info(df)
+    return df
+
+
+def clean_locations(df):
+    pass_end_location = df['pass.end_location']
+    shot_end_location = df['shot.end_location']
+    return df.sort_values(by=['match_id']).assign(**{
+        'location.x': df.location.map(lambda l: l[0], na_action='ignore'),
+        'location.y': df.location.map(lambda l: l[1], na_action='ignore'),
+        'pass.end_location.x': pass_end_location.map(lambda l: l[0], na_action='ignore'),
+        'pass.end_location.y': pass_end_location.map(lambda l: l[1], na_action='ignore'),
+        'shot.end_location.x': shot_end_location.map(lambda l: l[0], na_action='ignore'),
+        'shot.end_location.y': shot_end_location.map(lambda l: l[1], na_action='ignore'),
+        'shot.end_location.z': shot_end_location.map(lambda l: l[2] if len(l) < 3 else np.NaN, na_action='ignore'),
+    })
+
+
+def goalkeeper_info(df):
+    raise NotImplementedError
+
+
+def shot_info(df):
+    raise NotImplementedError
+
+
+def freeze_frame_info(df):
+    raise NotImplementedError
+
+
+def format_elapsed_time(df):
+    raise NotImplementedError
+
+
+def possession_info(df):
+    raise NotImplementedError