remain df index

espinielli · Feb 19, 2024 · 871e98a · 871e98a
1 parent f33c5a8
commit 871e98a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/fastmeteo/grid.py b/fastmeteo/grid.py
@@ -91,6 +91,8 @@ def sync_local(self, start: np.datetime64, stop: np.datetime64):
 
  def interpolate(self, flight: pd.DataFrame) -> pd.DataFrame:
  times = pd.to_datetime(flight.timestamp)
+ index = flight.index
+
  flight = flight.reset_index(drop=True).assign(
  longitude_360=lambda d: d.longitude % 360
  )
@@ -132,7 +134,10 @@ def interpolate(self, flight: pd.DataFrame) -> pd.DataFrame:
  ds.coords, method="linear", assume_sorted=False
  ).to_dataframe()[self.features]
 
- flight_new = pd.concat([flight, new_params], axis=1).drop(
- columns="longitude_360"
+ flight_new = (
+ pd.concat([flight, new_params], axis=1)
+ .drop(columns="longitude_360")
+ .set_index(index)
  )
+
  return flight_new