11 lines
447 B
Python
11 lines
447 B
Python
|
|
def populate_indicators(self, dataframe: pd.DataFrame, metadata: dict) -> pd.DataFrame:
|
|
import numpy as np
|
|
|
|
dataframe['ema_short'] = dataframe['close'].ewm(span=self.ema_short_period.value, adjust=False).mean()
|
|
dataframe['ema_long'] = dataframe['close'].ewm(span=self.ema_long_period.value, adjust=False).mean()
|
|
|
|
dataframe['rsi'] = ta.RSI(np.array(dataframe['close']), timeperiod=14)
|
|
|
|
return dataframe
|