Group Based Trajectory Modeling Python