mgbam commited on
Commit
3065e91
·
verified ·
1 Parent(s): 61d6ca7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -43,9 +43,14 @@ data_load_state.text("Loading SPY data...done!")
43
 
44
  # Calculate technical indicators: MACD and RSI
45
  macd_indicator = MACD(close=data['Close'])
46
- # Use flatten() to convert to a 1D array
47
- data['MACD'] = pd.Series(np.array(macd_indicator.macd()).flatten(), index=data.index)
48
- data['MACD_signal'] = pd.Series(np.array(macd_indicator.macd_signal()).flatten(), index=data.index)
 
 
 
 
 
49
 
50
  rsi_indicator = RSIIndicator(close=data['Close'], window=14)
51
  data['RSI'] = rsi_indicator.rsi()
 
43
 
44
  # Calculate technical indicators: MACD and RSI
45
  macd_indicator = MACD(close=data['Close'])
46
+
47
+ # Convert outputs to 1-dimensional lists
48
+ macd_values = np.array(macd_indicator.macd()).flatten().tolist()
49
+ macd_signal_values = np.array(macd_indicator.macd_signal()).flatten().tolist()
50
+
51
+ # Create the Series using the lists
52
+ data['MACD'] = pd.Series(macd_values, index=data.index)
53
+ data['MACD_signal'] = pd.Series(macd_signal_values, index=data.index)
54
 
55
  rsi_indicator = RSIIndicator(close=data['Close'], window=14)
56
  data['RSI'] = rsi_indicator.rsi()