0vergeared commited on
Commit
8d5c542
·
verified ·
1 Parent(s): 2afeeb0

Update indicators.py

Browse files
Files changed (1) hide show
  1. indicators.py +3 -3
indicators.py CHANGED
@@ -12,9 +12,6 @@ def calculate_rsi(df, period=14):
12
  rsi = 100 - (100 / (1 + rs))
13
  return rsi
14
 
15
- def calculate_ema(df, period=20):
16
- return df['close'].ewm(span=period, adjust=False).mean()
17
-
18
  def calculate_macd(df):
19
  ema12 = df['close'].ewm(span=12, adjust=False).mean()
20
  ema26 = df['close'].ewm(span=26, adjust=False).mean()
@@ -26,6 +23,9 @@ def generate_signal(df):
26
  rsi = calculate_rsi(df)
27
  macd, signal = calculate_macd(df)
28
 
 
 
 
29
  latest_rsi = rsi.iloc[-1]
30
  latest_macd = macd.iloc[-1]
31
  latest_signal = signal.iloc[-1]
 
12
  rsi = 100 - (100 / (1 + rs))
13
  return rsi
14
 
 
 
 
15
  def calculate_macd(df):
16
  ema12 = df['close'].ewm(span=12, adjust=False).mean()
17
  ema26 = df['close'].ewm(span=26, adjust=False).mean()
 
23
  rsi = calculate_rsi(df)
24
  macd, signal = calculate_macd(df)
25
 
26
+ if len(rsi) < 1 or len(macd) < 1 or len(signal) < 1:
27
+ return "Not enough data"
28
+
29
  latest_rsi = rsi.iloc[-1]
30
  latest_macd = macd.iloc[-1]
31
  latest_signal = signal.iloc[-1]