automate-stock-screener-2.4.py

# function to check for EMA Bounce
def check_bounce_EMA(df):
  candle1 = df.iloc[-1]
  candle2 = df.iloc[-2]
  cond1 = candle1['EMA18'] > candle1['EMA50'] > candle1['EMA100']
  cond2 = candle1['STOCH_%K(5,3,3)'] <= 30 or candle1['STOCH_%D(5,3,3)'] <= 30
  cond3 = candle2['Low'] < candle2['EMA50'] and \
          candle2['Close'] > candle2['EMA50'] and \
          candle1['Low'] > candle1 ['EMA50']
  return cond1 and cond2 and cond3

# price_chart_df is from Section 2.3
print(check_bounce_EMA(price_chart_df))