Files
OpenBB/openbb_terminal/stocks/quantitative_analysis/qa_model.py
Colin Delahunty 459e103a5a Fixes some James bugs (#2969)
* Fixed bug

* Fix vwap

* Hid future warning

* Added fix to pandas

* Quality improvement to customer and supplier

* Fixed some tests

* Test fix
2022-10-21 08:17:29 -04:00

15 lines
447 B
Python

import pandas as pd
import numpy as np
pd.options.mode.chained_assignment = None
def full_stock_df(df: pd.DataFrame) -> pd.DataFrame:
df["Returns"] = df["Adj Close"].pct_change()
df["LogRet"] = np.log(df["Adj Close"]) - np.log(df["Adj Close"].shift(1))
df["LogPrice"] = np.log(df["Adj Close"])
df = df.rename(columns={"Adj Close": "AdjClose"})
df = df.dropna()
df.columns = [x.lower() for x in df.columns]
return df