Final methods of the compound trajectory class

CompoundTrajectory_final.py

    @classmethod
    def _trim_array(cls, array: np.ndarray) -> np.ndarray:
        if array.shape[0] > 2:
            return array[1:-1]
        else:
            return array

    def to_trajectory(self) -> Trajectory:
        lat = np.concatenate([self._trim_array(segment.lat) for segment in self.segments])
        lon = np.concatenate([self._trim_array(segment.lon) for segment in self.segments])
        time = np.concatenate([self._trim_array(segment.time) for segment in self.segments])

        # Remove duplicates
        diff_lat = np.where(np.diff(lat) == 0)[0]

        lat = np.delete(lat, diff_lat)
        lon = np.delete(lon, diff_lat)
        time = np.delete(time, diff_lat)

        return Trajectory(lat=lat, lon=lon, time=time)