Python 量化投资原来这么简单(6) — 交易平均收益率

在之前的五篇系列文章中，计算收益率时没有考虑每次按策略进行交易时的收益率，而是单纯回测一段时间后，通过计算最终价值和本金的差距并除以本金，得到最终收益率。

这样的计算方法其实是不准确的，因为交易时每次都采用100股的形式进行，在没有引入调仓技术前，我们应该以每次交易的平均收益率为准。

具体计算方法如下：

• 在每次买入股票的时候，记录购买价格：self.buyprice

• 在每次卖出股票的时候，计算收益率：(卖出价格-买入价格)/买入价格，忽略佣金。

• 将每次交易收益存入params变量中，以便后续分析。

1.准备

开始之前，你要确保Python和pip已经成功安装在电脑上噢，如果没有，请访问这篇文章：超详细Python安装指南 进行安装。如果你用Python的目的是数据分析，可以直接安装Anaconda：Python数据分析与挖掘好帮手—Anaconda

Windows环境下打开Cmd(开始—运行—CMD)，苹果系统环境下请打开Terminal(command+空格输入Terminal)，准备开始输入命令安装依赖。

当然，我更推荐大家用VSCode编辑器，把本文代码Copy下来，在编辑器下方的终端运行命令安装依赖模块，多舒服的一件事啊：Python 编程的最好搭档—VSCode 详细指南。

在终端输入以下命令安装我们所需要的依赖模块:

pip install backtrader

看到 Successfully installed xxx 则说明安装成功。

Backtrader基本使用请看我们第一篇文章：

backtrader教程—量化投资原来这么简单(1)

本文全部代码，请在Python实用宝典后台回复：量化投资6 进行下载。

2.交易收益率

以第二篇的macd策略为例，首先初始化策略变量，用于记录股票交易的每次收益率：

第二，要在notify_order函数中记录购买时的价格：

第三，在卖出的时候，根据卖出价格和买入价格计算收益率，并存入策略变量里的profits变量中：

在策略运行完毕后，可以通过以下变量获得策略变量里profits的值：

cerebro.runstrats[0][0].params.profits

结果如下：

[0.021676761236850372, -0.1054225992123598, 0.10678571428571423, 0.11044953855314062, 0.21502209131075103, 0.15729837813819164, 0.10841304881039966, 0.16918294849023086]

可以看到，每次该策略进行交易时的收益率，第一次盈利2%、第二次亏损10%、第三次盈利10%、第四次盈利11%...

有趣的是第五次，盈利了21%，这个策略明明是在盈利10%的时候卖出，为什么这次交易能盈利21%？

主要是在涨停板之下，股票无法卖出导致的。

这也给我们一个hint：这个策略写死的涨/跌10%卖出是不是过于死板了，有没有更合适的卖出策略？

大家可以尝试一下找到更好的卖出策略，下一篇文章中我们将重点讨论这个问题。

3. 平均收益率

使用numpy的mean函数，可以直接算出数组的平均值：

print(np.mean(profits))

结果如下：

0.09792573520161482

平均下来，该策略在这只股票上的平均收益率为9.7%. 这个结果要比前几篇文章的计算方法更加准确，也更具有价值。

如果你喜欢今天的Python 教程，请持续关注Python实用宝典，如果对你有帮助，麻烦在下面点一个赞/在看，有任何问题都可以在下方留言，我们会耐心解答的！

点击下方阅读原文可以获取所有代码和链接哦！

Python实用宝典 (pythondict.com)

不只是一个宝典

欢迎关注公众号：Python实用宝典