添加新的周期数据

This commit is contained in:
jackyu66git
2026-02-02 00:35:51 +08:00
parent dd5ba33c0f
commit 42dac07274
5 changed files with 64 additions and 203 deletions
+11 -1
View File
@@ -22,7 +22,17 @@ class TF_DF():
def init_TF_DF(self, df, interval, timeframe):
self.timeframe = timeframe
self.interval = interval
self.dataframe = resample_to_interval(df, interval)
# 检查 DataFrame 是否为空或没有 date 列
if df is None or df.empty:
raise ValueError(f"DataFrame for {timeframe} is empty. Please download data first.")
if 'date' not in df.columns:
raise ValueError(f"DataFrame for {timeframe} missing 'date' column. Columns: {df.columns.tolist()}")
# interval=1 时不需要重采样
if interval == 1:
self.dataframe = df.copy()
else:
self.dataframe = resample_to_interval(df, interval)
#print(self.timeframe, len(self.dataframe))
self.dataframe = self.add_indicators(self.dataframe)
self.klu_list = []
self.klc_list = []