feat: add moving average plot

src/safeds/data/tabular/plotting/_table_plotter.py

@@ -353,6 +353,41 @@ def scatter_plot(self, x_name: str, y_names: list[str]) -> Image:
 
         return _figure_to_image(fig)
 
+    def moving_average_plot(self, x_name: str, y_name: str, window_size: int) -> Image:
+        import matplotlib.pyplot as plt
+        import numpy as np
+        import polars as pl
+
+        _plot_validation(self._table, x_name, [y_name])
+        # Calculate the moving average
+        mean_col = pl.col(y_name).mean().alias(y_name)
+        grouped = self._table._lazy_frame.sort(x_name).group_by(x_name).agg(mean_col).collect()
+        data = grouped
+        moving_average = data.select([pl.col(y_name).rolling_mean(window_size).alias("moving_average")])
+        # set up the arrays for plotting
+        y_data_with_nan = moving_average["moving_average"].to_numpy()
+        nan_mask = ~np.isnan(y_data_with_nan)
+        y_data = y_data_with_nan[nan_mask]
+        x_data = data[x_name].to_numpy()[nan_mask]
+        fig, ax = plt.subplots()
+        ax.plot(x_data, y_data, label="moving average")
+        ax.set(
+            xlabel=x_name,
+            ylabel=y_name,
+        )
+        ax.legend()
+        if self._table.get_column(x_name).is_temporal:
+            ax.set_xticks(x_data)  # Set x-ticks to the x data points
+        ax.set_xticks(ax.get_xticks())
+        ax.set_xticklabels(
+            ax.get_xticklabels(),
+            rotation=45,
+            horizontalalignment="right",
+        )  # rotate the labels of the x Axis to prevent the chance of overlapping of the labels
+        fig.tight_layout()
+
+        return _figure_to_image(fig)
+
 
 def _plot_validation(table: Table, x_name: str, y_names: list[str]) -> None:
     y_names.append(x_name)

tests/safeds/data/tabular/plotting/test_moving_average_plot.py

@@ -0,0 +1,58 @@
+import datetime
+
+import pytest
+from safeds.data.tabular.containers import Table
+from syrupy import SnapshotAssertion
+
+
+@pytest.mark.parametrize(
+    ("table", "x_name", "y_name", "window_size"),
+    [
+        (Table({"A": [1, 2, 3], "B": [2, 4, 7]}), "A", "B", 2),
+        # (Table({"A": [1, 1, 2, 2, 3, 3, 4, 4, 5, 5], "B": [2, 4, 6, 8, 10, 12, 14, 16, 18, 20]}), "A", "B", 2),
+        (
+            Table(
+                {
+                    "time": [
+                        datetime.date(2022, 1, 10),
+                        datetime.date(2022, 1, 10),
+                        datetime.date(2022, 1, 11),
+                        datetime.date(2022, 1, 11),
+                        datetime.date(2022, 1, 12),
+                        datetime.date(2022, 1, 12),
+                    ],
+                    "A": [10, 5, 20, 2, 1, 1],
+                },
+            ),
+            "time",
+            "A",
+            2,
+        ),
+        (
+            Table(
+                {
+                    "time": [
+                        datetime.date(2022, 1, 9),
+                        datetime.date(2022, 1, 10),
+                        datetime.date(2022, 1, 11),
+                        datetime.date(2022, 1, 12),
+                    ],
+                    "A": [10, 5, 20, 2],
+                },
+            ),
+            "time",
+            "A",
+            2,
+        ),
+    ],
+    ids=["numerical", "date grouped", "date"],
+)
+def test_should_match_snapshot(
+    table: Table,
+    x_name: str,
+    y_name: str,
+    window_size: int,
+    snapshot_png_image: SnapshotAssertion,
+) -> None:
+    line_plot = table.plot.moving_average_plot(x_name, y_name, window_size)
+    assert line_plot == snapshot_png_image