gravity = pd.read_csv('./gravity.csv')
gravity['gravity'] = gravity['dist_to_defender'] + gravity['teammates_distances']
gravity_on = gravity[gravity['is_dribbling']]
colsToDrop = ['possession_id', 'is_dribbling', 'dist_to_defender', 'teammates_distances']
gravity_metric = gravity_on.drop(colsToDrop, axis=1)
gravity_metric = gravity_metric.groupby('player_id').agg(np.mean)
print(gravity_metric)