本文作者: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询问。
如果你想入门量化,但是始终找不到方向,可以加入我的知识星球。我会在里面解答你的问题,分享我的感悟,不论是投资、技术,还是职业选择、思维方式。
-- 学习和成长从来都不是一个人的事 --
领取专属 10元无门槛券
私享最新 技术干货