Python進階量化交易場外篇2——線性迴歸擬合股價沉浮
新年伊始,很榮幸筆者的 《教你用 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、股票交易資料視覺化:技術分析常用指標繪製》小節的基礎上把移動線性迴歸擬合曲線增加到顯示介面中,箭頭所指示的位置作為買賣點目前來看是有一定的盈利特徵的,不過此處只是為了擴充套件大家的思路達到教學目的而設定的策略,大家可以在這個基礎上展開更多股票和週期的回測以驗證策略的可靠性。更多的量化交易內容歡迎大家訂閱專欄閱讀!!
更多python量化交易內容互動請加微信公眾號:PythonQT-YuanXiao