jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-27 12:40:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve sma plotting

Browse Source
This commit is contained in:
didier
2021-02-07 22:10:36 +00:00
parent c71c53edc3
commit 5eae9ea773
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
prediction_techniques/sma.py
View File

@@ -27,6 +27,7 @@ def simple_moving_average(l_args, s_ticker, s_interval, df_stock):
print(f"The following args couldn't be interpreted: {l_unknown_args}\n")
return
# Prediction data
l_predictions = list()
for pred_day in range(ns_parser.n_days):
if pred_day < ns_parser.n_length:
@@ -38,21 +39,22 @@ def simple_moving_average(l_args, s_ticker, s_interval, df_stock):
l_pred_days = get_next_stock_market_days(last_stock_day=df_stock['4. close'].index[-1], n_next_days=ns_parser.n_days)
df_pred = pd.Series(l_predictions, index=l_pred_days, name='Price')
# Plotting
plt.plot(df_stock.index, df_stock['4. close'], lw=2)
plt.title(f"{ns_parser.n_length} Moving Average on {s_ticker} - {ns_parser.n_days} days prediction")
plt.xlim(df_stock.index[0], get_next_stock_market_days(df_pred.index[-1], 10)[-1])
plt.xlim(df_stock.index[0], get_next_stock_market_days(df_pred.index[-1], 1)[-1])
plt.xlabel('Time')
plt.ylabel('Share Price ($)')
plt.grid(b=True, which='major', color='#666666', linestyle='-')
plt.minorticks_on()
plt.grid(b=True, which='minor', color='#999999', linestyle='-', alpha=0.2)
df_ma = df_stock['4. close'].rolling(window=ns_parser.n_length).mean()
plt.plot(df_ma.index, df_ma, lw=2)
plt.plot(df_pred.index, df_pred, lw=2)
cmap = plt.get_cmap("tab10")
plt.axvspan(df_pred.index[0], df_pred.index[-1], facecolor=cmap(1), alpha=0.2)
plt.plot(df_ma.index, df_ma, lw=2, linestyle='--', c='tab:orange')
plt.plot([df_stock.index[-1], df_pred.index[0]], [df_stock['4. close'].values[-1], df_pred.values[0]], lw=1, c='tab:green', linestyle='--')
plt.plot(df_pred.index, df_pred, lw=2, c='tab:green')
plt.axvspan(df_stock.index[-1], df_pred.index[-1], facecolor='tab:orange', alpha=0.2)
xmin, xmax, ymin, ymax = plt.axis()
plt.vlines(df_stock.index[-1], ymin, ymax, colors=cmap(0), linewidth=1, linestyle='--', color='k')
plt.vlines(df_stock.index[-1], ymin, ymax, linewidth=1, linestyle='--', color='k')
plt.show()
df_pred = df_pred.apply(lambda x: f"{x:.2f} $")