Files
OpenBB/openbb_terminal/stocks/quantitative_analysis/qa_model.py
Colin Delahunty ee34a1f278 Feature/add isort linter (#4114)
* Fixed isort

* Fixed issues

* FIxed linters

---------

Co-authored-by: James Maslek <jmaslek11@gmail.com>
2023-02-04 21:24:47 -05:00

15 lines
447 B
Python

import numpy as np
import pandas as pd
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