mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
Added export to stocks/qa/beta (#2510)
* added export to beta * added alpha and beta to filename and removed it from the actual file
This commit is contained in:
@@ -2,12 +2,14 @@
|
||||
__docformat__ = "numpy"
|
||||
|
||||
import logging
|
||||
import os
|
||||
|
||||
import matplotlib.pyplot as plt
|
||||
import pandas as pd
|
||||
from openbb_terminal.decorators import log_start_end
|
||||
from openbb_terminal.rich_config import console
|
||||
from openbb_terminal.stocks.quantitative_analysis.beta_model import beta_model
|
||||
from openbb_terminal.helper_funcs import export_data
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -18,6 +20,7 @@ def beta_view(
|
||||
ref_symbol: str,
|
||||
data: pd.DataFrame = None,
|
||||
ref_data: pd.DataFrame = None,
|
||||
export: str = "",
|
||||
) -> None:
|
||||
"""Display the beta scatterplot + linear regression.
|
||||
|
||||
@@ -52,3 +55,12 @@ def beta_view(
|
||||
ax.text(0.9, 0.1, beta_text, horizontalalignment="right", transform=ax.transAxes)
|
||||
fig.show()
|
||||
console.print()
|
||||
|
||||
df = pd.DataFrame({"sr": sr, "rr": rr})
|
||||
|
||||
export_data(
|
||||
export,
|
||||
os.path.dirname(os.path.abspath(__file__)),
|
||||
f"beta_alpha={alpha}_beta={beta}",
|
||||
df,
|
||||
)
|
||||
|
||||
@@ -783,7 +783,9 @@ class QaController(StockBaseController):
|
||||
parser, other_args, EXPORT_ONLY_RAW_DATA_ALLOWED
|
||||
)
|
||||
if ns_parser:
|
||||
beta_view(self.ticker, ns_parser.ref, data=self.stock)
|
||||
beta_view(
|
||||
self.ticker, ns_parser.ref, data=self.stock, export=ns_parser.export
|
||||
)
|
||||
|
||||
@log_start_end(log=logger)
|
||||
def call_var(self, other_args: List[str]):
|
||||
|
||||
Reference in New Issue
Block a user