mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-03 07:51:49 +00:00
ee34a1f278
* Fixed isort * Fixed issues * FIxed linters --------- Co-authored-by: James Maslek <jmaslek11@gmail.com>
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
""" Relative Strength Percentile Model """
|
|
__docformat__ = "numpy"
|
|
|
|
import logging
|
|
from typing import Tuple
|
|
|
|
import pandas as pd
|
|
|
|
from openbb_terminal.decorators import log_start_end
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def get_rsp(
|
|
s_ticker: str = "",
|
|
) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, pd.DataFrame]:
|
|
"""Relative strength percentile [Source: https://github.com/skyte/relative-strength]
|
|
Currently takes from https://github.com/soggyomelette/rs-log in order to get desired output
|
|
|
|
Parameters
|
|
----------
|
|
s_ticker : str
|
|
Stock Ticker
|
|
|
|
Returns
|
|
-------
|
|
Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, pd.DataFrame]
|
|
Dataframe of stock percentile, Dataframe of industry percentile,
|
|
Raw stock dataframe for export, Raw industry dataframe for export
|
|
"""
|
|
|
|
df_stock_p = pd.read_csv(
|
|
"https://raw.githubusercontent.com/soggyomelette/rs-log/main/output/rs_stocks.csv"
|
|
)
|
|
df_industries_p = pd.read_csv(
|
|
"https://raw.githubusercontent.com/soggyomelette/rs-log/main/output/rs_industries.csv"
|
|
)
|
|
rsp_stock = pd.DataFrame()
|
|
rsp_industry = pd.DataFrame()
|
|
|
|
if s_ticker != "":
|
|
rsp_stock = df_stock_p[df_stock_p["Ticker"].str.match(s_ticker)]
|
|
for i in range(len(df_industries_p)):
|
|
if s_ticker in df_industries_p.iloc[i]["Tickers"]:
|
|
rsp_industry = df_industries_p.iloc[[i]]
|
|
|
|
return (rsp_stock, rsp_industry, df_stock_p, df_industries_p) # type: ignore
|