1
0

增加结构化策略

This commit is contained in:
2025-01-15 18:43:23 +08:00
parent a89c60227c
commit b74e33b37b
4 changed files with 139 additions and 9 deletions

15
selector.py Normal file
View File

@@ -0,0 +1,15 @@
import pandas as pd
class Selector:
def select(self, codes: [str], df: pd.DataFrame) -> [str]:
return codes
class PeriodSelector(Selector):
def __init__(self, period: int = 5):
self.__period = period
def select(self, codes: [str], df: pd.DataFrame) -> [str]:
size_df = df.groupby("code").size()
return list(filter(lambda code: size_df[code] > self.__period, codes))