Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
wiseplat
GitHub Repository: wiseplat/python-code
Path: blob/master/ invest-robot-contest_trading_bot-master/trading/candles/add_indicators.py
5935 views
1
from ta.trend import ema_indicator, macd, adx
2
3
4
'''
5
Тут представлены все функции, которые позволяют добавить в DataFrame необходимые индикааторы
6
для дальнейшего использования в анализе.
7
'''
8
9
'''
10
Функция для добавления индикатора EMA
11
'''
12
13
14
def add_ema(df, window=7):
15
16
df[f"ema_{window}"] = ema_indicator(close=df['close'], window=window)
17
18
return df
19
20
21
'''
22
Функция для добавления индикатора MACD
23
'''
24
25
26
def add_macd(df):
27
28
df['macd'] = macd(close=df['close'])
29
30
return df
31
32
33
'''
34
Функция для добавления индикатора ADX
35
'''
36
37
38
def add_adx(df, window=14):
39
40
df['adx'] = adx(close=df['close'], high=df['high'], low=df['low'], window=window)
41
42
return df
43