veles/marketTrade
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

revised dataloader

Browse Source
This commit is contained in:
Mark 2022-06-04 16:32:02 +03:00
parent c608d73f07
commit aaee1a9fcf
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
market_trade/dataloader.py
View File

@ -8,9 +8,13 @@ class DukaMTInterface:
has two attributes -- bids and asks has two attributes -- bids and asks
""" """
def __init__(self, duka_candlesticks_filepath): def __init__(self, duka_candlesticks):
# reading index from column 0 and header takes two rows, because of multirow indexers # checking if provided is path or pandas dataframe
self.duka_dataset = pd.read_csv(duka_candlesticks_filepath, index_col=0, header=[0, 1], parse_dates=True) if isinstance(duka_candlesticks, pd.DataFrame):
self.duka_dataset = duka_candlesticks
else:
# reading index from column 0 and header takes two rows, because of multirow indexers
self.duka_dataset = pd.read_csv(duka_candlesticks, index_col=0, header=[0, 1], parse_dates=True)
# renaming timestamp to date # renaming timestamp to date
self.duka_dataset['date'] = self.duka_dataset.index self.duka_dataset['date'] = self.duka_dataset.index