Bugfix in the dataframe unpacking function which would not flatten al…

…l locations.
ecmwf-projects · Nov 13, 2023 · b649ce0 · b649ce0
1 parent c56e88e
commit b649ce0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ibicus/utils/_utils.py b/ibicus/utils/_utils.py
@@ -461,9 +461,9 @@ def _unpack_df_of_numpy_arrays(df, numpy_column_name):
     new_expanded_rows = []
     for _, row in df.iterrows():
         expanded_row = {}
-        for index, value in row.iteritems():
+        for index, value in row.items():
             if index == numpy_column_name:
-                expanded_row[index] = value[0].flatten()
+                expanded_row[index] = value.flatten()
             else:
                 expanded_row[index] = value
         new_expanded_rows.append(pd.DataFrame(data=expanded_row))