create_timeseries.reshape_unevenly()

Complexity

Conditions

4

Size

Total Lines	23
Code Lines	16

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

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