diff --git a/CHANGELOG.md b/CHANGELOG.md index 1e2ac21663..8d0c316ec0 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -12,6 +12,7 @@ but cannot always guarantee backwards compatibility. Changes that may **break co **Improved** - New model: `StatsForecastAutoTBATS`. This model offers the [AutoTBATS](https://nixtlaverse.nixtla.io/statsforecast/src/core/models.html#autotbats) model from Nixtla's `statsforecasts` library. [#2611](https://github.com/unit8co/darts/pull/2611) by [He Weilin](https://github.com/cnhwl). +- Added the `title` attribute to `TimeSeries.plot()`. This allows to set a title for the plot. [#2639](https://github.com/unit8co/darts/pull/2639) by [Jonathan Koch](https://github.com/jonathankoch99). - Added parameter `component_wise` to `show_anomalies()` to separately plot each component in multivariate series. [#2544](https://github.com/unit8co/darts/pull/2544) by [He Weilin](https://github.com/cnhwl). **Fixed** diff --git a/darts/timeseries.py b/darts/timeseries.py index 6f2597dd15..4b7940e91f 100644 --- a/darts/timeseries.py +++ b/darts/timeseries.py @@ -4077,6 +4077,7 @@ def plot( low_quantile: Optional[float] = 0.05, high_quantile: Optional[float] = 0.95, default_formatting: bool = True, + title: Optional[str] = None, label: Optional[Union[str, Sequence[str]]] = "", max_nr_components: int = 10, ax: Optional[matplotlib.axes.Axes] = None, @@ -4106,6 +4107,8 @@ def plot( interval is shown if `high_quantile` is None (default 0.95). default_formatting Whether to use the darts default scheme. + title + Optionally, a custom plot title. If `None`, will use the name of the underlying `xarray.DataArray`. label Can either be a string or list of strings. If a string and the series only has a single component, it is used as the label for that component. If a string and the series has multiple components, it is used as @@ -4254,7 +4257,7 @@ def plot( ) ax.legend() - ax.set_title(self._xa.name) + ax.set_title(title if title is not None else self._xa.name) return ax def with_columns_renamed(