Add seasonal_plot

[Mr-Geekman]

IMPORTANT: Please do not create a Pull Request without creating an issue first.

Before submitting (must do checklist)

• Did you read the contribution guide?
• Did you update the docs? We use Numpy format for all the methods and classes.
• Did you write any new necessary tests?
• Did you update the CHANGELOG?

Type of Change

• Examples / docs / tutorials / contributors update
• Bug fix (non-breaking change which fixes an issue)
• Improvement (non-breaking change which improves an existing feature)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)

Proposed Changes

Look #548.

Related Issue

#548.

Closing issues

Closes #548.

[Mr-Geekman]

Example script:

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

from etna.analysis import seasonal_plot
from etna.datasets import TSDataset


def main():
    df = pd.read_csv("examples/data/example_dataset.csv", parse_dates=["timestamp"])
    df_wide = TSDataset.to_dataset(df)
    df_wide.iloc[:10, 0] = np.NaN
    ts = TSDataset(df=df_wide, freq="D")

    # make plot
    seasonal_plot(ts=ts, freq=None, cycle="month")
    plt.savefig("seasonal_plot_day_month")

    seasonal_plot(ts=ts, freq=None, cycle="year")
    plt.savefig("seasonal_plot_day_year")

    seasonal_plot(ts=ts, freq=None, cycle="week", figsize=(10, 10))
    plt.savefig("seasonal_plot_day_week")

    seasonal_plot(ts=ts, freq="W", cycle="year", aggregation="mean")
    plt.savefig("seasonal_plot_week_year_mean")

    seasonal_plot(ts=ts, freq="M", cycle="year", aggregation="sum")
    plt.savefig("seasonal_plot_month_year_sum")

    seasonal_plot(ts=ts, freq=None, cycle=50, alignment="first")
    plt.savefig("seasonal_plot_day_50_first")

    seasonal_plot(ts=ts, freq=None, cycle=50, alignment="last")
    plt.savefig("seasonal_plot_day_50_last")


if __name__ == "__main__":
    main()

seasonal_plot_day_month:

seasonal_plot_day_year:

seasonal_plot_day_week:

seasonal_plot_week_year_mean:

seasonal_plot_month_year_sum:

seasonal_plot_day_50_first:

seasonal_plot_day_50_last:

[codecov-commenter]

Codecov Report

Merging #628 (4dd72e5) into master (ae3095c) will increase coverage by 0.24%.
The diff coverage is 96.72%.

@@            Coverage Diff             @@
##           master     #628      +/-   ##
==========================================
+ Coverage   84.01%   84.26%   +0.24%     
==========================================
  Files         118      118              
  Lines        6120     6242     +122     
==========================================
+ Hits         5142     5260     +118     
- Misses        978      982       +4     

Impacted Files	Coverage Δ
etna/analysis/eda_utils.py	48.82% <96.61%> (+31.50%)	⬆️
etna/analysis/__init__.py	100.00% <100.00%> (ø)

📣 Codecov can now indicate which changes are the most critical in Pull Requests. Learn more

[martins0n]

Yеt another problem with duplicated legend, isn't it?

[martins0n]

🤯

[martins0n]

dict with key of type Tuple[SeasonalPlotCycle, str]?

[martins0n]

🥇