|
| 1 | +# To add a new cell, type '# %%' |
| 2 | +# To add a new markdown cell, type '# %% [markdown]' |
| 3 | + |
| 4 | +# %% |
| 5 | +import yfinance as yf |
| 6 | +import pandas_datareader as pdr |
| 7 | +from mpl_finance import candlestick_ohlc |
| 8 | +import matplotlib.pyplot as plt |
| 9 | + |
| 10 | + |
| 11 | +# %% |
| 12 | +# import SPY stock price |
| 13 | +df_spy = pdr.get_data_yahoo("SPY", start="2019-01-01", end="2019-09-30") |
| 14 | +# import AAPL stock price |
| 15 | +df_aapl = pdr.get_data_yahoo("AAPL", start="2019-01-01", end="2019-09-30") |
| 16 | + |
| 17 | + |
| 18 | +# %% |
| 19 | +df_spy.head() |
| 20 | + |
| 21 | + |
| 22 | +# %% |
| 23 | +df_aapl[["Open", "High", "Low", "Close"]].plot() |
| 24 | +plt.show() |
| 25 | + |
| 26 | + |
| 27 | +# %% |
| 28 | +fig = plt.figure(figsize=(10, 10)) |
| 29 | +ax = plt.subplot() |
| 30 | + |
| 31 | +plot_data = [] |
| 32 | +for i in range(150, len(df_aapl)): |
| 33 | + row = [ |
| 34 | + i, |
| 35 | + df_aapl.Open.iloc[i], |
| 36 | + df_aapl.High.iloc[i], |
| 37 | + df_aapl.Low.iloc[i], |
| 38 | + df_aapl.Close.iloc[i], |
| 39 | + ] |
| 40 | + plot_data.append(row) |
| 41 | +candlestick_ohlc(ax, plot_data) |
| 42 | +plt.show() |
| 43 | + |
| 44 | + |
| 45 | +# %% |
| 46 | +from stockstats import StockDataFrame |
| 47 | +stocks = StockDataFrame.retype(df_aapl[["Open", "Close", "High", "Low", "Volume"]]) |
| 48 | + |
| 49 | + |
| 50 | +# %% |
| 51 | +plt.plot(stocks["close_10_sma"], color="b", label="SMA") |
| 52 | +plt.plot(df_aapl.Close, color="g", label="Close prices") |
| 53 | +plt.legend(loc="lower right") |
| 54 | +plt.show() |
| 55 | + |
| 56 | + |
| 57 | +# %% |
| 58 | +plt.plot(stocks["close_10_sma"], color="b", label="SMA") # plotting SMA |
| 59 | +plt.plot(stocks["close_10_ema"], color="k", label="EMA") |
| 60 | +plt.plot(df_aapl.Close, color="g", label="Close prices") # plotting close prices |
| 61 | +plt.legend(loc="lower right") |
| 62 | +plt.show() |
| 63 | + |
| 64 | + |
| 65 | +# %% |
| 66 | +plt.plot(stocks["macd"], color="b", label="MACD") |
| 67 | +plt.plot(stocks["macds"], color="g", label="Signal Line") |
| 68 | +plt.legend(loc="lower right") |
| 69 | +plt.show() |
| 70 | + |
| 71 | + |
0 commit comments