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() # 获取股票代码列表
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],
"turn": item_row[9],
"pctChg": item_row[10]
}
dm.add_tk_item(ticker, __dict) # 将数据保存到数据库
2.编写交易逻辑
为了便于描述,本文选择了一个较为简单的交易逻辑。以周为交易周期,每周一开盘前分析各股的周macd数据,满足交易条件则以开盘价买入并持有一周,再以当周五的收盘价卖出。这个交易逻辑比较简单且实操性强,回测的结果也有可圈可点之处(回测结果见文末)。交易逻辑的核心代码如下:
if wmacd_list[-1] > 0 >= wmacd_list[-2]: # 判断某支股票是否符合当前交易逻辑
if np.mean(volume_list[-5:-1]) < volume_list[-1]:
if 0.1 >= diff_list[-1] >= 0:
data = [x for x in dm_tk.find_one_by_key({"ticker": item["ticker"]})["data_list"] if x["date"] == cur_date][0]
result_list.append(data)
def get_w_macd(price_list): # 生成每支股票的周macd数据
ema_12_list = list()
for index in range(len(price_list)):
if index == 0:
ema_12_list.append(price_list[0])
else:
ema_12_list.append(round(ema_12_list[index - 1] * 11 / 13 + price_list[index] * 2 / 13, 4))
ema_26_list = list()
for index in range(len(price_list)):
if index == 0:
ema_26_list.append(price_list[0])
else:
ema_26_list.append(round(ema_26_list[index - 1] * 25 / 27 + price_list[index] * 2 / 27, 4))
diff_list = list()
for index in range(len(ema_12_list)):
diff = ema_12_list[index] - ema_26_list[index]
diff_list.append(diff)
dea_list = list()
for index in range(len(diff_list)):
if index == 0:
dea_list.append(diff_list[0])
else:
dea_list.append(round(dea_list[index - 1] * 0.8 + diff_list[index] * 0.2, 4))
wmacd_list = list()
for index in range(len(dea_list)):
bar = (diff_list[index] - dea_list[index]) * 3
wmacd_list.append(bar)
return wmacd_list, diff_list, dea_list
以上只是该交易策略的部分代码,读者不需要看懂其中的逻辑,实际操作过程中应该使用自己的交易策略。
3.模拟交易操作
编写好交易策略后,我们开始对交易策略进行回测。首先我们设定一些初始数据,这些数据是我们日常交易中常见到的,比如初始资金总额、当前可用资金、最大仓位等等,由于该交易策略比较简单,所以我们只需要设定起始资金就可以了:
capital_base = 1000000 # 起始资金设定为100万
history_capital = list() # 用于记录交易结果
接着我们创建一条时间轴,所有的交易操作都将跟随时间轴进行:
# 生成时间轴
def date_range(start, end, step=1, format="%Y-%m-%d"):
strptime, strftime = datetime.datetime.strptime, datetime.datetime.strftime
days = (strptime(end, format) - strptime(start, format)).days + 1
return [strftime(strptime(start, format) + datetime.timedelta(i), format) for i in range(0, days, step)]
date_list = date_range("2016-01-01", "2016-12-31") # 生成2016-01-01至2016-12-31的所有时间点
生成好时间轴后,使用for循环遍历时间轴(模拟时间推进,并且过滤掉周末和节假日),按照之前设定的交易策略,我们在每周一和周五进行买入卖出操作即可。由于该策略不涉及加减仓,故我们对交易过程进行了简化,通过直接计算得出每周的收益。
对于更为复杂的交易策略,建议开发者分别实现开仓、平仓和加减仓等各种操作:
for cur_date in date_list:
if datetime.datetime.strptime(cur_date, "%Y-%m-%d").weekday() == 4: # 判断当前日期是否需要操作
result_list = list() # 用于记录当前时间符合交易条件的股票代码
for item in code_list: # 遍历各支股票,筛选出符合交易条件的股票
-执行交易逻辑-
if 符合交易条件:
result_list.append(data)
# 计算本次操作的收益
if result_list:
capital = capital_base / len(result_list) # 对当前资金进行均分
temp_capital = 0
for item in result_list:
close_price = float(item["close"])
open_price = float(item["open"])
max_price = float(item["high"])
profit = (close_price - open_price) / open_price
temp_capital += (capital * (1 + profit))
capital_base = temp_capital
history_capital.append(capital_base) # 记录本次操作后剩余的资金
4.统计结果和绘图
模拟交易完成后我们来对结果进行统计,由于我们已经将交易的过程记录在history_capital中,此时我们可以轻松的计算出收益率:
net_rate = (history_capital[-1] - history_capital[0]) / history_capital[0] # 计算回测结果
log.logger.info("total_profit:" + str(round(net_rate * 100, 2)) + "%")
为了让交易的结果更加直观,我们还可以将其绘制成折线图,这里使用matplotlib进行绘图:
plt.subplot(111)
lable_x = np.arange(len(history_capital))
plt.plot(lable_x, history_capital, color="r", linewidth=1.0, linestyle="-")
plt.xlim(lable_x.min(), lable_x.max() * 1.1)
plt.ylim(min(history_capital) * 0.9, max(history_capital) * 1.1)
plt.grid(True)
plt.show()
回测结果展示(该收益曲线是在限制最大仓位的条件下得出的,如果取消该限制,收益率将更高):
2014年全年收益
2015年全年收益
2016年全年收益
2017年全年收益
到此为止,我们就完成了对某个交易策略进行回测的全部流程,从回测结果中可以看出,该交易策略在2014-2017这4年中有着不错的表现。按照策略的规则,交易者只需要在每周一开盘前运行策略,开盘后买进策略推荐的股票,最后在周五收盘前卖掉即可,是易于实盘操作的。
读者也可以自由变换交易逻辑来获取不同的结果,通过对回测结果进行分析,可以对我们日常的交易带来一些帮助。
与我交流:1003882179@qq.com
python股票回测_Python量化交易-回测简单的交易策略相关推荐
- python股票编程入门_Python股票量化投资-3.python基础
Python股票量化投资-1.开发环境部署 Python股票量化投资-2.量化投资介绍 继续开始今天的内容,主要介绍 PyCharm的开发使用[这IDE对JAVA人员来说不陌生] Python的语法推 ...
- python抓取交易所_python爬取上海石油天然气交易中心液化天然气交易数据并可视化分析...
由于之前从事过石化行业,上海石油天然气交易中心成立虽较晚(2015年成立),但一直有关注,去年下半年由于大量供暖采取煤改气,还一个没有霾笼罩的冬天,因此出现了天然气大幅涨价供不应求的状况,闲着没事就用 ...
- python股票成交明细_Python股票成交价格-买卖额分布图(三)
在Python股票气泡图(一) - 知乎专栏介绍了股票成交by每天by每价格的画图,它能够给你一个直观的每天成交价格.成交量的分布. 在Python股票成交价格分布图(二) - 知乎专栏介绍了股票成交 ...
- 下标索引必须为正整数类型或逻辑类型_python量化基础 | 变量和简单的数据类型,零基础都可以看懂...
编辑 | Cowboy 校对 | 李明 来源 | 牛角财经 目的 | python量化基础 | 变量和简单的数据类型,零基础都可以看懂!!! python教程 从入门到高级(免费) 特点:案例基于金融 ...
- python jieba库分词_Python基于jieba库进行简单分词及词云功能实现方法
本文实例讲述了Python基于jieba库进行简单分词及词云功能实现方法.分享给大家供大家参考,具体如下: 目标: 1.导入一个文本文件 2.使用jieba对文本进行分词 3.使用wordcloud包 ...
- 期货交易python回测_python量化交易历史回测
import tushare as ts import xlrd import pandas as pd import matplotlib.pyplot as plt ts.set_token('* ...
- python怎么检验股票日收益率_Python量化笔记-股票收益率的正态分布检验和凯利公式应用...
本为继承上一篇: 完成以下扩展练习: 4.2 扩展练习2:对股票的收益率进行正态分布检验 4.3 扩展练习3:如果股票的收益率是正态分布的,使用凯利公式进行每日交易 4.2 扩展练习2:对股票的收益率 ...
- python判断字符串回文_python实现--判断回文字符串、回文链表、回文数
所谓回文字符串,就是正读和反读都一样的字符串,比如"level"或者"noon"等等就是回文串.即是对称结构 Python系列教程,免费获取,遇到bug及时反馈 ...
- python画蜡烛图_Python量化交易-绘制蜡烛图 !这个图不像你的钱哦!
mpl_finance模块已经从matlibplot里独立出来,实现了蜡烛线绘制功能,其包含的函数有: 1. candlestick2_ochl(ax,opens,closes,highs,lows, ...
最新文章
- MVC系列框架之Struts存在的意义是什么?
- HiSDP —— 高效的C++软件开发平台
- SAP 开发陷阱一箩筐(01)——创建了函数组却无法激活
- Response.Redirect、 Server.Transfer、Server.Execute三者区别
- java中封装日期加时间_java日期处理简单封装
- 如何美化linux终端输出
- android数据序列化的实现
- python是什么类型的编程语言-python是一种什么类型的编程语言
- Tensorflow:模型训练tensorflow.train
- [k8s]容器化node-expolore(9100)+cadvisor(8080)+prometheus(9090) metric搜集,grafana展示
- html5视频播放器 知乎,iPhone、iPad 如何播放网页调用优酷视频?
- ps 毛发 边缘_【常用抠图小技巧】PS调整边缘扣毛发
- 如何使用Java以编程方式在Excel中创建数据透视表?
- 京东白条数据架构进化之路:要在数据的不确定性中探索架构的稳定性
- 远程ntp服务器响应模式6查询,H3C WP5048无线PoE注入器 命令参考-Release 2208-6W100
- openFOAM中的Scalar
- java运行时加音乐_Java运行窗体/程序添加自定义背景音乐
- 编程应该怎么学?学好编程讲方法
- git@gitlab.com: Permission denied报错
- 小程序真正实现多个文件上传,不通过循环调用uni.uploadFile