netflypsb commited on
Commit
89bad28
1 Parent(s): 3298018

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +71 -0
app.py ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import numpy as np
4
+ import matplotlib.pyplot as plt
5
+ import yfinance as yf
6
+
7
+ @st.cache_data
8
+ def load_data(ticker):
9
+ # Fetch data from Yahoo Finance
10
+ return yf.download(ticker, start="2000-01-01", end="2023-01-01")
11
+
12
+ ticker = st.text_input("Enter the ticker symbol", "AAPL")
13
+ data = load_data(ticker)
14
+
15
+ st.title("Algorithmic Trading Strategy Backtesting")
16
+
17
+ # Moving Average Windows
18
+ short_window = st.number_input("Short moving average window", 1, 50, 20)
19
+ long_window = st.number_input("Long moving average window", 1, 200, 50)
20
+
21
+ # Initial Capital
22
+ initial_capital = st.number_input("Initial Capital", 1000, 1000000, 100000)
23
+
24
+ # Calculate moving averages
25
+ data['Short_MA'] = data['Close'].rolling(window=short_window).mean()
26
+ data['Long_MA'] = data['Close'].rolling(window=long_window).mean()
27
+
28
+ # Drop NaN values
29
+ data.dropna(inplace=True)
30
+
31
+ # Generate signals
32
+ data['Signal'] = 0
33
+ data['Signal'][short_window:] = np.where(data['Short_MA'][short_window:] > data['Long_MA'][short_window:], 1, 0)
34
+ data['Position'] = data['Signal'].diff()
35
+
36
+ # Show signals in data
37
+ st.write(data.tail())
38
+
39
+ # Simulate portfolio
40
+ data['Portfolio Value'] = initial_capital
41
+ data['Portfolio Value'][short_window:] = initial_capital * (1 + data['Signal'][short_window:].shift(1) * data['Close'].pct_change()[short_window:]).cumprod()
42
+
43
+ # Performance metrics
44
+ cagr = (data['Portfolio Value'].iloc[-1] / initial_capital) ** (1 / ((data.index[-1] - data.index[short_window]).days / 365.25)) - 1
45
+ sharpe_ratio = data['Portfolio Value'].pct_change().mean() / data['Portfolio Value'].pct_change().std() * np.sqrt(252)
46
+
47
+ st.write(f"CAGR: {cagr:.2%}")
48
+ st.write(f"Sharpe Ratio: {sharpe_ratio:.2f}")
49
+
50
+ # Plot strategy performance
51
+ plt.figure(figsize=(10, 5))
52
+ plt.plot(data.index, data['Portfolio Value'], label='Portfolio Value')
53
+ plt.title(f"Backtested Performance of {ticker} Strategy")
54
+ plt.xlabel("Date")
55
+ plt.ylabel("Portfolio Value")
56
+ plt.legend()
57
+ st.pyplot()
58
+
59
+ # Highlight buy and sell signals
60
+ fig, ax = plt.subplots(figsize=(10, 5))
61
+ ax.plot(data.index, data['Close'], label='Close Price', alpha=0.5)
62
+ ax.plot(data.index, data['Short_MA'], label=f'Short MA ({short_window})', alpha=0.75)
63
+ ax.plot(data.index, data['Long_MA'], label=f'Long MA ({long_window})', alpha=0.75)
64
+ ax.plot(data[data['Position'] == 1].index, data['Short_MA'][data['Position'] == 1], '^', markersize=10, color='g', lw=0, label='Buy Signal')
65
+ ax.plot(data[data['Position'] == -1].index, data['Short_MA'][data['Position'] == -1], 'v', markersize=10, color='r', lw=0, label='Sell Signal')
66
+ plt.title(f"{ticker} Price and Trading Signals")
67
+ plt.xlabel("Date")
68
+ plt.ylabel("Price")
69
+ plt.legend()
70
+ st.pyplot(fig)
71
+