Update indicators.py
Browse files- indicators.py +3 -2
indicators.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
-
import
|
| 2 |
|
| 3 |
def calculate_ema(df, window):
|
| 4 |
-
|
|
|
|
| 5 |
return df
|
|
|
|
| 1 |
+
from ta.trend import EMAIndicator
|
| 2 |
|
| 3 |
def calculate_ema(df, window):
|
| 4 |
+
ema = EMAIndicator(close=df['Close'], window=window)
|
| 5 |
+
df[f"ema_{window}"] = ema.ema_indicator()
|
| 6 |
return df
|