Merge pull request #79 from CIAT-DAPA/develop

add function to sum year
CIAT-DAPA · Oct 17, 2023 · 817af5a · 817af5a
2 parents edd150d + 164331a
commit 817af5a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/aclimate_resampling/resampling.py b/src/aclimate_resampling/resampling.py
@@ -525,8 +525,15 @@ def save_forecast(self, station, output_root, year_forecast, seasons_range, base
       year_forecast = int(year_forecast)
 
       for i in range(len(IDs)):
+           #         print(IDs[i])
+          df = seasons_range[(seasons_range['id'] == IDs[i])]
+
+          df = df.reset_index()
+          df = df.drop(columns = ['year'])
+
+          for j in list(range(len(df))):
 
-          df.loc[j, 'year'] = self.add_year(year_forecast, df.loc[j, 'month'])
+              df.loc[j, 'year'] = self.add_year(year_forecast, df.loc[j, 'month'])
 
  #