量化小讲堂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询问。