jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-15 13:51:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

add start time to stats plots

Browse Source
This commit is contained in:
didier
2021-03-27 18:03:28 +00:00
parent c68327ee32
commit b64fcf8359
2 changed files with 19 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
gamestonk_terminal/exploratory_data_analysis/eda_api.py
View File

@@ -5,6 +5,7 @@ import argparse
from typing import List
from detecta import detect_cusum
from matplotlib import pyplot as plt
from datetime import datetime
import pandas as pd
from pandas.plotting import register_matplotlib_converters
import numpy as np
@@ -60,7 +61,7 @@ def summary(other_args: List[str], stock: pd.DataFrame):
return
def hist(other_args: List[str], ticker: str, stock: pd.DataFrame):
def hist(other_args: List[str], ticker: str, stock: pd.DataFrame, start: datetime):
"""Plot histogram and density
Parameters
@@ -100,7 +101,9 @@ def hist(other_args: List[str], ticker: str, stock: pd.DataFrame):
rug=True,
rug_kws={"edgecolor": "orange"},
)
plt.title(ticker + " (Histogram with Density)")
plt.title(
f"Histogram with Density of {ticker} from {start.strftime('%Y-%m-%d')}"
)
plt.ylabel("Density")
plt.xlabel("Share Price")
plt.grid(True)
@@ -116,7 +119,7 @@ def hist(other_args: List[str], ticker: str, stock: pd.DataFrame):
return
def cdf(other_args: List[str], ticker: str, stock: pd.DataFrame):
def cdf(other_args: List[str], ticker: str, stock: pd.DataFrame, start: datetime):
"""Plot cumulative distribution function
Parameters
@@ -147,7 +150,9 @@ def cdf(other_args: List[str], ticker: str, stock: pd.DataFrame):
cdf = stock.value_counts().sort_index().div(len(stock)).cumsum()
cdf.plot(lw=2)
plt.title(ticker + " (Cumulative Distribution Function)")
plt.title(
f"Cumulative Distribution Function of {ticker} from {start.strftime('%Y-%m-%d')}"
)
plt.ylabel("Probability")
plt.xlabel("Share Price")
minVal = stock.values.min()
@@ -198,7 +203,7 @@ def cdf(other_args: List[str], ticker: str, stock: pd.DataFrame):
return
def bwy(other_args: List[str], ticker: str, stock: pd.DataFrame):
def bwy(other_args: List[str], ticker: str, stock: pd.DataFrame, start: datetime):
"""Box and Whisker plot yearly
Parameters
@@ -230,7 +235,9 @@ def bwy(other_args: List[str], ticker: str, stock: pd.DataFrame):
sns.set(style="whitegrid")
box_plot = sns.boxplot(x=stock.index.year, y=stock)
box_plot.set(
xlabel="Year", ylabel="Share Price", title=ticker + " (Box-plot per Year)"
xlabel="Year",
ylabel="Share Price",
title=f"Box-plot per Year of {ticker} from {start.strftime('%Y-%m-%d')}",
)
if gtff.USE_ION:
@@ -244,7 +251,7 @@ def bwy(other_args: List[str], ticker: str, stock: pd.DataFrame):
return
def bwm(other_args: List[str], ticker: str, stock: pd.DataFrame):
def bwm(other_args: List[str], ticker: str, stock: pd.DataFrame, start: datetime):
"""Box and Whisker plot monthly
Parameters
@@ -278,7 +285,7 @@ def bwm(other_args: List[str], ticker: str, stock: pd.DataFrame):
box_plot.set(
xlabel="Month",
ylabel="Share Price",
title=ticker + " (Box-plot per Month)",
title=f"Box-plot per Month of {ticker} from {start.strftime('%Y-%m-%d')}",
)
l_months = [
"Jan",

8
gamestonk_terminal/exploratory_data_analysis/eda_controller.py
View File

@@ -113,19 +113,19 @@ class EdaController:
def call_hist(self, other_args: List[str]):
"""Process hist command"""
eda_api.hist(other_args, self.ticker, self.stock)
eda_api.hist(other_args, self.ticker, self.stock, self.start)
def call_cdf(self, other_args: List[str]):
"""Process cdf command"""
eda_api.cdf(other_args, self.ticker, self.stock)
eda_api.cdf(other_args, self.ticker, self.stock, self.start)
def call_bwy(self, other_args: List[str]):
"""Process bwy command"""
eda_api.bwy(other_args, self.ticker, self.stock)
eda_api.bwy(other_args, self.ticker, self.stock, self.start)
def call_bwm(self, other_args: List[str]):
"""Process bwm command"""
eda_api.bwm(other_args, self.ticker, self.stock)
eda_api.bwm(other_args, self.ticker, self.stock, self.start)
def call_rolling(self, other_args: List[str]):
"""Process rolling command"""