Python量化交易线性回归拟合股价沉浮[程序化老手]

转自：segmentfault
新年伊始，很荣幸笔者的《教你用 Python 进阶量化交易》专栏在慕课专栏板块上线了，欢迎大家订阅！为了能够提供给大家更轻松的学习过程，笔者在专栏内容之外会陆续推出一些手记来辅助同学们学习本专栏内容，因此同学们无需担心专栏内容在学习上的困难，更多的是明确自己学习的目的即可。当然笔者也欢迎同学们踊跃留言，说出自己想扩展的知识点，笔者会根据同学们的意愿选择性的推出一些内容。

本次场外篇笔者在专栏《22、股票交易策略开发：走势线性回归选股策略》的基础上对线性回归方法的策略应用做进一步的扩展介绍，同时会涉及到多个小节的知识内容，大家可以参考专栏内容。

这里我们顺带介绍下搭建环境中可能会遇到的问题，如果有同学是用mac系统开发的，在MAC中调试matplotlib时中文显示框框解决方法：
1、下载simhei.ttf字体库拷贝至matplotlib字体文件夹（Macintosh HD ▸ 用户 ▸ SHQ ▸ anaconda3 ▸ lib ▸ python3.7 ▸ site-packages ▸ matplotlib ▸ mpl-data ▸ fonts▸ ttf）。
2、同样在matplotlib/mpl-data/fonts目录下面修改配置文件matplotlibrc 
font.family : sans-serif
font.sans-serif : SimHei, Bitstream Vera Sans, Lucida Grande, Verdana, Geneva, Lucid, Arial, Helvetica, Avant Garde, sans-serif
axes.unicode_minus:False 
3、在Python中运行如下代码重新加载字体使配置生效：
from matplotlib.font_manager import _rebuild
_rebuild() #reload

言归正传，在22小节中笔者选取了浙大网新、高鸿股份、开山股份、水晶光电四只股票的收盘价以线性回归的方式拟合出走势的角度，从而评估它们未来的趋势。

由于线性回归作用于股票收盘价的整个周期，因此选择合适的时间段非常重要。比如以高鸿股份为例，以下是它从2018年1月1日至今的走势图（参照股票《16、股票交易数据可视化：技术分析常用指标绘制》实现）。

从图中可以看到2018年4月10月是单边下降趋势，而从10月份开始转为上升趋势。那么我们分别对这两段周期做线性回归后发现，单边下降通道的拟合直线角度为-0.97度，而上升通道拟合直线角度为1.5度，股价的拟合直线角度从-0.97度至1.5度的转变过程，其实也是由跌转涨的过程。


在制定量化交易策略时有一种方式是寻找现象中的特征，根据特征去生成交易条件。那么这里给大家一个衍生的策略思路，既然线性回归作用于股票收盘价的整个周期，前后两段完全相反的周期会彼此作用，最终影响拟合的角度值，我们可以设定窗口期用移动窗口的方式拟合股票的走势。

在22小节的FactorPickStockAng类中增加一个方法，代码如下：

def fit_price(self, symbols, cycle=30, show=True):
        
    for index, stockName in enumerate(symbols.keys()):
            
        #kl_pd = getdata #参考fit_pick()方法，此处略

        kl_pd.fillna(method='bfill', inplace=True) 
           
        for kl_index in np.arange(0, kl_pd.shape[0]):
                
            if kl_index >= cycle:
                    
                kl_pd.loc[kl_pd.index[kl_index], "ang"], x, reg_y_fit, y_arr = self.calc_regress_deg(kl_pd.Close[kl_index-cycle:kl_index])                 
                print(kl_pd.index[kl_index],kl_pd.iloc[kl_index, kl_pd.columns.get_loc("ang")])
            
        if show:
                
            kl_pd.fillna(method='bfill', inplace=True)    
            plt.plot(np.arange(0, len(kl_pd.index)), kl_pd.ang)
  
     plt.title('compare ang')
        
     plt.legend(symbols.keys(), loc='best')
        
     plt.show()

执行该代码examp_trade.fit_price(pick_stocks)可生成浙大网新、高鸿股份、开山股份、水晶光电四只股票的收盘价移动线性回归曲线图，可以发现角度曲线的拐点预示开始反转为新一轮的走势。

在《16、股票交易数据可视化：技术分析常用指标绘制》小节的基础上把移动线性回归拟合曲线增加到显示界面中，箭头所指示的位置作为买卖点目前来看是有一定的盈利特征的，不过此处只是为了扩展大家的思路达到教学目的而设定的策略，大家可以在这个基础上展开更多股票和周期的回测以验证策略的可靠性。更多的量化交易内容欢迎大家订阅专栏阅读！！