请问python如何做股票交易回测？

1.获取证券数据

创建策略之前我们要明确一个问题，这个交易策略将会被用在哪个市场？这样我们才能有针对性的获取数据。本文以A股市场为例，先获取A股近10年的数据并保存到数据库。

1..编写数据库操作类

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

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)

# 添加历史交易记录
def add_tk_item(self, ticker, __dict):
return self.table.update_one({'ticker': ticker}, {"$push": {"data_list": __dict}})

1.2.获取数据 获取证券数据的途径分为两种，第一种是去网上找现成的数据接口，通过调用接口获取数据，这种方式的优点在于简单便捷，数据的准确性有保障；第二种是自己编写数据爬虫获取数据，这种方式会相对麻烦一点，但如果你对数据的时效性要求较高，或者数据接口中的数据不能满足你的需求，可以选择第二种方式。本文采用的是第一种方式。使用的数据接口是http://www.baostock.com/。 调用接口：

bs.login() # 初始化baostock
code_list = dm.get_code_list() # 获取股票代码列表
for item in code_list:
max_try = 8 # 失败重连的最大次数
ticker = item["ticker"]
for tries in range(max_try):
rs = bs.query_history_k_data(ticker, "date,code,open,high,low,close,volume,amount,adjustflag,turn,"
"pctChg", frequency="w", adjustflag="3")
if rs.error_code == '0':
parse_pager(rs, ticker) # 解析数据
break
elif tries < (max_try - 1):
sleep(2)
continue
else:
log.logger.error("加载数据失败：" + str(ticker))
log.logger.info("加载数据完成")
bs.logout()

解析数据：

# 解析数据并保存到数据库
def parse_pager(content, ticker):
while content.next():
item_row = content.get_row_data()
__dict = {
"date": item_row[0],
"code": item_row[1],
"open": item_row[2],
"high": item_row[3],
"low": item_row[4],
"close": item_row[5],
"volume": item_row[6],
"amount": item_row[7],
"adjustflag": item_row[8],
"tur...

点击查看剩余70%