Fix extraction of chart data

empowerplan · Nov 20, 2024 · 4153803 · 4153803
1 parent 23cc9b6
commit 4153803
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/digiplan/map/charts.py b/digiplan/map/charts.py
@@ -50,7 +50,11 @@ def render(self) -> dict:
                 self.chart_options["series"][0]["data"] = data
             elif series_length > 1:
                 for i in range(0, series_length):
-                    values = self.chart_data.iloc[i]
+                    values = (
+                        self.chart_data.iloc[i]
+                        if isinstance(self.chart_data, (pd.DataFrame, pd.Series))
+                        else self.chart_data[i]
+                    )
                     if not isinstance(values, (list, tuple)):
                         values = [values]
                     self.chart_options["series"][i]["data"] = values