fix issue
This commit is contained in:
parent
aee39f7f69
commit
295f071e1e
@ -245,6 +245,12 @@ class FreqaiPrimer(IStrategy):
|
||||
|
||||
# 获取15m数据进行趋势确认
|
||||
df_15m = self.dp.get_pair_dataframe(pair=metadata['pair'], timeframe='15m')
|
||||
|
||||
# 确保df_15m中包含rsi_15m列,如果没有则计算
|
||||
if 'rsi_15m' not in df_15m.columns:
|
||||
# 使用talib计算14期RSI
|
||||
df_15m['rsi_15m'] = self.talib.RSI(df_15m['close'], timeperiod=14)
|
||||
|
||||
df_15m = df_15m.rename(columns={'date': 'date_15m'})
|
||||
merged_data = dataframe.merge(df_15m[['date_15m', 'rsi_15m']], how='left', left_on='date', right_on='date_15m')
|
||||
merged_data = merged_data.fillna(method='ffill')
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user