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

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

批量爬取股票数据

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

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

#上证代码

shanghaicode = []

for i in range(600000, 604000, 1):

shanghaicode.append(str(i))

#深证代码

shenzhencode = []

for i in range(1000000, 1005000, 1):

i = str(i)[1:]

shenzhencode.append(i)

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

def getalldata(code):

if os.path.exists(datapath + code + '.csv'):

print(code + 'already existed!')

return

metadata = broker.get_stock_pro(code)

if len(metadata) == 0:

return

metadata.to_csv('C:/Users/abc/Desktop/' + code + '.