量化小讲堂34:万能Python-交易策略买点、卖点可视化

引言:

邢不行的系列帖子“量化小讲堂”,通过实际案例教初学者使用python进行量化投资,了解行业研究方向,希望能对大家有帮助。

个人微信:coinquant,有问题欢迎交流

万能Python | 交易策略买点、卖点可视化

本文作者:Arthur

可视化一直是数据挖掘以及机器学习中常用的办法,当然它可以应用到方方面面,比如交易策略买卖信号的展示。

如下图所示,向上的红色箭头代表做多,向下的红色箭头代表做空,绿色圆点代表平仓。

绘图效果,点击图片放大查看

这篇文章介绍怎么画出这样的图,方便我们直观地看策略的运行机理。文章代码由邢不行(个人微信coinquant)的学生Arthur提供。

01

想要画出上图,只需要在数字货币Python量化投资课程 | 邢不行,第八课中的【案例1】择时策略.py 的文件中找到这一行代码:

这行代码的作用:删除不需要的列

在这一行代码之前,加入绘图的代码。最终这一段代码是这样的:

整段代码

然后运行程序,之后会跳出一个小窗口:

绘图代码运行的结果

这就是我们画出的图。图中红色的三角形代表开仓的信号,三角形向上代表做多,向下代表做空;绿色的圆形代表平仓的信号。

02

这幅图目前并不是很美观,也看不清楚。不过幸好,这个小窗口具有交互性。选中工具栏中的拖动按键,就可以拖动整幅图:

拖动按键

选中放大镜,用鼠标左键选择一个区域,就是放大选中的区域。用鼠标右键选择一个区域,就是缩小整个图片。

文章开头的图片所展示的,就是放大之后的原图。在放大图片之后,还可以用拖动工具,仔细观察每一段放大后的图片。

可拖动,可放大缩小,自由调整比例

03

如果想了解这幅图的制作原理,继续阅读:

首先,python 中有一个包,叫做 matplotlib。其中,最常用的一个工具是 matplotlib 下面的 pyplot,这个工具可以画出很多具有交互功能的图形。

很多工具包的画图功能也是通过 pyplot 构建起来的,例如 pandas.DataFrame.plot,pandas.Series.plot,还有 seaborn。我们可以使用 pyplot 与这些画图工具配合画出一张图。

我们逐行解释一下这段绘图代码讲了一些什么。

代码作用逐行讲解

建议大家在 pycharm 的 console 窗口中逐行运行上述脚本,如果直接使用 run 脚本的方式,记得不要省略最后的 plt.show()。

虽然复制粘贴就可以运行,还是建议大家手打一遍,印象更深刻一些。

最后,感谢 Arthur 为大家提供的这篇教程。

需要文中的代码,可以加邢不行的微信coinquant询问。

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20181107A0PXYN00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券