import numpy as np def detect_bounce(trajectory): ys = [y for _, y in trajectory] dy = np.diff(ys) bounce_index = np.argmax(dy > 15) if np.any(dy > 15) else None return int(bounce_index) if bounce_index is not None else None