python实现马丁策略回测3000只股票的实例代码

时间:2022-10-12 08:09:07

上一篇文章讲解了如何实现马丁策略,但没有探索其泛化能力,所以这次来尝试回测3000只股票来查看盈利比例。

批量爬取股票数据

这里爬取数据继续使用tushare,根据股票代码来遍历,因为爬取数据需要一定时间,不妨使用多线程来爬取,这里要注意tushare规定每分钟爬取不能超过500次,除非你有很多积分,所以线程数要适当调低。

首先我们生成上证与深证所有股票的代码:

  1. #上证代码
  2. shanghaicode = []
  3. for i in range(600000, 604000, 1):
  4. shanghaicode.append(str(i))
  5.  
  6. #深证代码
  7. shenzhencode = []
  8. for i in range(1000000, 1005000, 1):
  9. i = str(i)[1:]
  10. shenzhencode.append(i)

然后再定义一个爬取函数,broker则是上一篇文章创建的实例:

  1. def getalldata(code):
  2. if os.path.exists(datapath + code + '.csv'):
  3. print(code + 'already existed!')
  4. return
  5. metadata = broker.get_stock_pro(code)
  6. if len(metadata) == 0:
  7. return
  8. metadata.to_csv('C:/Users/abc/Desktop/' + code + '.csv',index = False)
  9. print(code + 'finished!')

导入多线程需要的模块

  1. from concurrent.futures.thread import ThreadPoolExecutor #多线程

遍历所有代码开始爬取,max_workers可适当调整

  1. executor = ThreadPoolExecutor(max_workers=3)
  2. for datatemp in executor.map(getalldata, shenzhencode):
  3. pass
  4.  
  5. executor = ThreadPoolExecutor(max_workers=3)
  6. for datatemp in executor.map(getalldata, shanghaicode):
  7. pass

批量回测股票

数据爬好后则可开始回测了,因为回测是CPU瓶颈运算,所以这里就不使用多线程了,速度差不多。

首先将一只股票的回测程序封装到函数中,回测时间设置为2020年全年,起始资金设置为20万元:

  1. def martinmulti(code):
  2. broker = backtesting(200000,'20200101', '20201231')
  3. #获取股票数据
  4. metadata = pd.read_csv(datapath + code)
  5. data = np.array(metadata['close'])
  6. exdata = np.array(metadata['pre_close'])
  7. everyChange = np.array(metadata['change'])
  8. date = metadata['trade_date'].values
  9. everyChange = everyChange/data
  10. #开始回测
  11. broker.startbackmartin(data, exdata, everyChange, date)
  12. dicttemp = {'股票代码': code,'终止现金': broker.cash}
  13. return dicttemp

遍历股票代码回测并记录终止现金

  1. cashlist = pd.DataFrame(columns= ['股票代码','终止现金'])
  2. for code in datalist:
  3. datatemp = martinmulti(code)
  4. cashlist = cashlist.append(datatemp,ignore_index=True)

回测过程如下

python实现马丁策略回测3000只股票的实例代码

接下来看看哪支股票获得了最大利润:

python实现马丁策略回测3000只股票的实例代码

看看平均值

  1. cashlist.mean()
  2. Out[12]:
  3. 终止现金 208279.115166

可以从均值看出马丁策略赚作为一种相对保险的方法赚的不多,当然想要找到一劳永逸的方法是不可能的,并且用平均数不能代表一切,那看看盈利比例如何:

  1. plt.rcParams['font.sans-serif'] = ['SimHei']
  2. plt.style.use('ggplot')
  3. plt.title("盈利分布(万元)")
  4. bins = []
  5. for i in range(10000, 600000, 10000):
  6. bins.append(i)
  7. plt.hist(cashlist['终止现金'],bins = bins)
  8. plt.axvline(x = cashlist.mean().values,ls="-",c="green")#添加垂直直线

python实现马丁策略回测3000只股票的实例代码

可以看出有折腰的也有翻倍的,且绝大部分集中于20w元旁边,分布图形整体往20万右侧偏移,该策略还有待改进。

到此这篇关于python实现马丁策略回测3000只股票的文章就介绍到这了,更多相关python股票策略回测内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://blog.csdn.net/weixin_40264579/article/details/112867522