Update strategy_analysis_example.md
A better approach to calculate equity and equity_daily
This commit is contained in:
@@ -178,20 +178,8 @@ import pandas as pd
|
|||||||
stats = load_backtest_stats(backtest_dir)
|
stats = load_backtest_stats(backtest_dir)
|
||||||
strategy_stats = stats['strategy'][strategy]
|
strategy_stats = stats['strategy'][strategy]
|
||||||
|
|
||||||
dates = []
|
df = pd.DataFrame(columns=['dates','equity'], data=strategy_stats['daily_profit'])
|
||||||
profits = []
|
df['equity_daily'] = df['equity'].cumsum()
|
||||||
for date_profit in strategy_stats['daily_profit']:
|
|
||||||
dates.append(date_profit[0])
|
|
||||||
profits.append(date_profit[1])
|
|
||||||
|
|
||||||
equity = 0
|
|
||||||
equity_daily = []
|
|
||||||
for daily_profit in profits:
|
|
||||||
equity_daily.append(equity)
|
|
||||||
equity += float(daily_profit)
|
|
||||||
|
|
||||||
|
|
||||||
df = pd.DataFrame({'dates': dates,'equity_daily': equity_daily})
|
|
||||||
|
|
||||||
fig = px.line(df, x="dates", y="equity_daily")
|
fig = px.line(df, x="dates", y="equity_daily")
|
||||||
fig.show()
|
fig.show()
|
||||||
|
|||||||
Reference in New Issue
Block a user