用matplotlib绘制每次交易的盈亏三角形

用matplotlib绘制每次交易的盈亏三角形

结果:

代码:

python

def plot_trade_triangle(self):
# plot each trade as a trade-triangle, and annotate pnl
trade = self.trade
equity = self.equity.equity
fig,ax=plt.subplots()
for dt, row in trade.iterrows():
bars = row.buybar, row['sellbar']
pnl = 0 , row['pnl']
colors = 'r' if row.pnl>0 else 'g' ax.plot( bars, pnl, color=colors)
arrowprops = dict(facecolor='black',
edgecolor=colors,
#shrink=0.05,
arrowstyle='->',)
y_factor = 1.2 if abs(row.pnl) > 10. else 1.6
colors = 'yellow' if row.pnl>0 else 'cyan'
valignment = 'bottom' if row['pnl']>0 else 'top'
note= 'hld:%s\npnl:%s'% (bars[1]-bars[0], str(row['pnl']))
ax.annotate(note, xy=(row.sellbar, row.pnl),
xytext=(row.sellbar, row.pnl*y_factor),
va=valignment,
arrowprops=arrowprops,)
ax.fill_between(bars, pnl, color=colors) # 画: 持股期间的浮动盈亏:
float_pnl=(roc1[row.buybar+1:row.sellbar+1]+1).cumprod()
iloc = [equity.index.get_loc(key) for key in float_pnl.index]
float_pnl = pd.Series( (float_pnl.values-1)*100., index=iloc)
ax.plot(float_pnl, color='b')
ax.set_ylabel('float_pnl(%)')