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

Make yahoo finance dataframe match index name of alpha vantage dataframe

Browse Source
This commit is contained in:
didier
2021-03-11 20:46:42 +00:00
parent ae8abd71f6
commit 264f91b16c
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gamestonk_terminal/main_helper.py
View File

@@ -183,6 +183,8 @@ def load(l_args, s_ticker, s_start, s_interval, df_stock):
"Volume": "6. volume",
}
)
df_stock.index.name = "date"
# Intraday
else:
if ns_parser.source == "av":
@@ -200,6 +202,7 @@ def load(l_args, s_ticker, s_start, s_interval, df_stock):
print(
"Unfortunately, yahoo finance historical data doesn't contain intraday prices"
)
return [s_ticker, s_start, s_interval, df_stock]
except Exception as e:
print(e)

5
gamestonk_terminal/prediction_techniques/fbprophet.py
View File

@@ -44,8 +44,8 @@ def fbprophet(l_args, s_ticker, df_stock):
df_stock = df_stock.sort_index(ascending=True)
df_stock.reset_index(level=0, inplace=True)
df_stock = df_stock[["Date", "5. adjusted close"]]
df_stock = df_stock.rename(columns={"Date": "ds", "5. adjusted close": "y"})
df_stock = df_stock[["date", "5. adjusted close"]]
df_stock = df_stock.rename(columns={"date": "ds", "5. adjusted close": "y"})
df_stock["ds"] = pd.to_datetime(df_stock["ds"])
model = Prophet(yearly_seasonality=False, daily_seasonality=False)
@@ -75,6 +75,7 @@ def fbprophet(l_args, s_ticker, df_stock):
facecolor="tab:orange",
alpha=0.2,
)
print("here3")
plt.ylim(ymin, ymax)
plt.xlim(
df_stock["ds"].values[0], get_next_stock_market_days(l_pred_days[-1], 1)[-1]