python股票回测_Python量化交易-回测简单的交易策略

这篇文章主要介绍如何使用Python对一些简单的交易策略进行回测，对这块比较感兴趣的朋友可以看一看。

1.获取证券数据

本文以A股市场为例，先获取A股近10年的数据并保存到数据库。

1.1.安装数据库（MongoDB）

为了提升运行效率，需要将证券数据保存到本地数据库，这里我们选择的数据库是MongoDB，安装过程在此不再赘述，参照http://www.runoob.com/mongodb/mongodb-window-install.html即可，比较简单。

1.2.编写数据库操作类

安装完数据库，我们先编写一个工具类来管理数据库的增删改查等操作：

class DBManager:

def __init__(self, table_name):

self.client = MongoClient("127.0.0.1", 27017)

self.db = self.client["my_database"]

self.table = self.db[table_name]

def clsoe_db(self):

self.client.close()

# 获取股票代码列表(sz格式)

def get_code_list(self):

return self.table.find({}, {"ticker": 1}, no_cursor_timeout=True)

# 查询多条数据

def find_by_key(self, request=None):

if request is None:

request = {}

return self.table.find(request)

# 查询单条数据

def find_one_by_key(self, request=None):

if request is None:

request = {}

return self.table.find_one(request)

# 添加单条数据

def add_one(self, post, created_time=datetime.datetime.now()):

# 添加一条数据

post['created_time'] = created_time

return self.table.insert_one(post)

1.3.获取数据

获取证券数据的途径主要有两种，第一种是去网上找现成的数据接口，通过调用接口获取数据，这种方式简单便捷，数据的准确性有保障；第二种是自己编写数据爬虫获取数据，这种方式会相对麻烦一点。本文采用的是第一种方式。使用的数据接口是http://www.baostock.com/。

调用数据接口：

bs.login() # 初始化baostock

code_list = dm.get_code_list() # 获取股票代码列表