From 146cc6b805e96cf2a04e458df4fd214628541708 Mon Sep 17 00:00:00 2001 From: Ben Daniels Date: Sat, 13 Feb 2021 18:31:07 -0500 Subject: [PATCH] Added timeseries to dataframes that return a date --- quiverquant.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/quiverquant.py b/quiverquant.py index 1df8608..e204822 100644 --- a/quiverquant.py +++ b/quiverquant.py @@ -22,6 +22,9 @@ def congress_trading(self, ticker="", politician=False): r = requests.get(url, headers=self.headers) j = json.loads(r.content) df = pd.DataFrame(j) + if not politician and len(ticker)>0: + df["Date"] = pd.to_datetime(df["Date"]) + df = df.set_index("Date") return df def gov_contracts(self, ticker=""): @@ -43,6 +46,10 @@ def lobbying(self, ticker=""): r = requests.get(url, headers=self.headers) df = pd.DataFrame(json.loads(r.content)) + if len(ticker) > 0: + # converts index of pandas dataframe to Date time index rather than a tuple + df["Date"] = pd.to_datetime(df["Date"]) + df = df.set_index("Date") return df @@ -58,6 +65,10 @@ def wikipedia(self, ticker=""): raise NameError('Upgrade your subscription plan to access this dataset.') df = pd.DataFrame(json.loads(r.content)) + if len(ticker) > 0: + # converts index of pandas dataframe to Date time index rather than a tuple + df["Date"] = pd.to_datetime(df["Date"]) + df = df.set_index("Date") return df def wallstreetbets(self, ticker=""): @@ -72,6 +83,10 @@ def wallstreetbets(self, ticker=""): raise NameError('Upgrade your subscription plan to access this dataset.') df = pd.DataFrame(json.loads(r.content)) + if len(ticker) > 0: + # converts index of pandas dataframe to Date time index rather than a tuple + df["Date"] = pd.to_datetime(df["Date"]) + df = df.set_index("Date") return df