mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-05 03:50:57 +08:00
17 lines
462 B
Python
17 lines
462 B
Python
import datetime
|
|
import pandas as pd
|
|
|
|
from qlib.data.inst_processor import InstProcessor
|
|
|
|
|
|
class ResampleProcessor(InstProcessor):
|
|
def __init__(self, freq: str, hour: int, minute: int):
|
|
self.freq = freq
|
|
self.hour = hour
|
|
self.minute = minute
|
|
|
|
def __call__(self, df: pd.DataFrame, *args, **kwargs):
|
|
df = df.loc[df.index.time == datetime.time(self.hour, self.minute)]
|
|
df.index = df.index.normalize()
|
|
return df
|