本次使用tushare接口,股票为600018,时间是20200101——20211231。
使用的指标是:收盘价大于简单移动平均价。
佣金为0.001。
英文代码的英文原文:Quickstart Guide - Backtrader
注意:代码最后plot画图,在spyder中不能正常显示图片,只能显示提示信息:
<IPython.core.display.Javascript object> <IPython.core.display.HTML object>
在Jupyter Notebook中则可以正常显示图片。
期初资金: 100000.00 2020-02-26, Close, 11.20 2020-02-26, 买入单, 11.20 2020-02-27, 已买入, 价格: 11.20, 费用: 112.00, 佣金 0.11 2020-02-27, Close, 11.21 2020-02-28, Close, 10.85 2020-02-28, 卖出单, 10.85 2020-03-02, 已卖出, 价格: 10.95, 费用: 112.00, 佣金 0.11 2020-03-02, 交易利润, 毛利润 -2.50, 净利润 -2.72 2020-03-02, Close, 11.04 2020-03-03, Close, 11.06 ...... 2022-01-21, Close, 8.76 2022-01-24, Close, 8.69 2022-01-25, Close, 8.56 2022-01-25, 卖出单, 8.56 2022-01-26, 已卖出, 价格: 8.57, 费用: 86.30, 佣金 0.09 2022-01-26, 交易利润, 毛利润 -0.60, 净利润 -0.77 2022-01-26, Close, 8.57 2022-01-27, Close, 8.45 2022-01-28, Close, 8.41 期末资金: 99968.07
from __future__ import (absolute_import, division, print_function, from datetime import datetime # For datetime objects # Import the backtrader platform class TestStrategy(bt.Strategy): def log(self, txt, dt=None): dt = dt or self.datas[0].datetime.date(0) print('%s, %s' % (dt.isoformat(), txt)) self.dataclose = self.datas[0].close # Add a MovingAverageSimple indicator self.sma = bt.indicators.SimpleMovingAverage( self.datas[0], period=self.params.maperiod) # Indicators for the plotting show bt.indicators.ExponentialMovingAverage(self.datas[0], period=25) bt.indicators.WeightedMovingAverage(self.datas[0], period=25, bt.indicators.StochasticSlow(self.datas[0]) bt.indicators.MACDHisto(self.datas[0]) rsi = bt.indicators.RSI(self.datas[0]) bt.indicators.SmoothedMovingAverage(rsi, period=10) bt.indicators.ATR(self.datas[0], plot=False) def notify_order(self, order): if order.status in [order.Submitted, order.Accepted]: # broker 提交/接受了,买/卖订单则什么都不做 if order.status in [order.Completed]: '已买入, 价格: %.2f, 费用: %.2f, 佣金 %.2f' % self.buyprice = order.executed.price self.buycomm = order.executed.comm self.log('已卖出, 价格: %.2f, 费用: %.2f, 佣金 %.2f' % self.bar_executed = len(self) elif order.status in [order.Canceled, order.Margin, order.Rejected]: self.log('订单取消/保证金不足/拒绝') def notify_trade(self, trade): self.log('交易利润, 毛利润 %.2f, 净利润 %.2f' % (trade.pnl, trade.pnlcomm)) self.log('Close, %.2f' % self.dataclose[0]) if self.dataclose[0] > self.sma[0]: self.log('买入单, %.2f' % self.dataclose[0]) if self.dataclose[0] < self.sma[0]: self.log('卖出单, %.2f' % self.dataclose[0]) def get_data(code,start='2020-01-01',end='2022-01-31'): df=ts.get_k_data(code,autype='qfq',start=start,end=end) df.index=pd.to_datetime(df.date) df=df[['open','high','low','close','volume','openinterest']] dataframe=get_data('600018') start=datetime(2020, 1, 1) end=datetime(2021, 12, 31) if __name__ == '__main__': data = bt.feeds.PandasData(dataname=dataframe, fromdate=start, todate=end) cerebro.addstrategy(TestStrategy) cerebro.broker.setcash(100000.0) cerebro.addsizer(bt.sizers.FixedSize, stake=10) cerebro.broker.setcommission(commission=0.001) print('期初资金: %.2f' % cerebro.broker.getvalue()) print('期末资金: %.2f' % cerebro.broker.getvalue())
|