adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adjust plot_dataframe to use btanalysis

Browse Source
This commit is contained in:
Matthias
2019-03-07 21:23:53 +01:00
parent e1f48c2b46
commit 9f7f089d8a
2 changed files with 16 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
freqtrade/data/btanalysis.py
View File

@@ -8,6 +8,10 @@ import pandas as pd
from freqtrade.misc import json_load
# must align with columns in backtest.py
BT_DATA_COLUMNS = ["pair", "profitperc", "open_time", "close_time", "index", "duration",
"open_rate", "close_rate", "open_at_end", "sell_reason"]
def load_backtest_data(filename) -> pd.DataFrame:
"""
@@ -24,11 +28,7 @@ def load_backtest_data(filename) -> pd.DataFrame:
with filename.open() as file:
data = json_load(file)
# must align with columns in backtest.py
columns = ["pair", "profitperc", "open_time", "close_time", "index", "duration",
"open_rate", "close_rate", "open_at_end", "sell_reason"]
df = pd.DataFrame(data, columns=columns)
df = pd.DataFrame(data, columns=BT_DATA_COLUMNS)
df['open_time'] = pd.to_datetime(df['open_time'],
unit='s',