create_timeseries

Complexity

Total Complexity

4

Size/Duplication

Total Lines	27
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	reshape_unevenly()	0	23	4

import pandas as pd


def reshape_unevenly(df):

    def to_bucket(ts: pd.Timestamp) -> pd.Timestamp:
        h = ts.hour
        d = ts.normalize()
        if 5 <= h <= 21:
            return ts
        if h in (21, 22, 23):
            return d + pd.Timedelta(hours=21)
        if h in (0,):
            return (d - pd.Timedelta(days=1)) + pd.Timedelta(hours=21)
        # h in (1, 2, 3, 4, 5)
        return d + pd.Timedelta(hours=3)

    buckets = df.index.map(to_bucket)
    buckets = buckets.where(buckets >= df.index[0], df.index[0])

    df_mean = df.groupby(buckets).mean().sort_index()
    # df_sum = df.groupby(buckets).sum().sort_index()
    df_mean.index.name = 'timestamp'
    # df_sum.index.name = 'timestamp'

    return df_mean