用Python快速分析和预测股票价格

点击上方“Python数据之道”，选择“星标公众号”

精品文章，第一时间送达

作者 | Vincent Tatan 

译者 | 李洁

编辑 | Lemonbit

出品 | Python数据之道

用Python快速分析、可视化和预测股票价格

1 前言

某天，我的一个朋友告诉我说，实现经济自由的关键是股票投资。虽然这是市场繁荣时期的真理，但如今业余交易股票仍然是一个有吸引力的选择。由于在线交易平台的便利性，涌现了许多自主价值投资者或家庭主妇交易员。甚至还有一些成功的故事和广告吹嘘有“快速致富计划”学习如何投资回报率高达 40% 甚至更高的股票。投资已成为当今职场人士的福音。

现在的问题是：哪些股票？如何分析股票？与其他股票相比，所选股票的回报和风险是什么？

本文的目标是让你了解使用快速简单的 Python 代码分析股票的一种方法。只需花 12 分钟阅读这篇文章——最好自己完成一下。然后你就可以快速浏览到你的第一份金融分析报告。

为了开始学习和分析股票，我们将从快速查看历史股票价格开始。这将通过从 Pandas 网络数据阅读器和雅虎财经中提取最新的股票数据来实现。然后，我们将尝试通过探索性分析，如相关性热图、Matplotlib 可视化以及使用线性分析和k最近邻（K Nearest Neighbor,KNN）的预测分析来查看数据。

2 加载雅虎财经数据集

Pandas 网络数据阅读器 （Pandas web data reader）是 Pandas 库的一个扩展，用于与大多数最新的金融数据进行通信，包括雅虎财经、谷歌财经、Enigma 等资源。

我们将使用以下代码提取 Apple 的股票价格：

import pandas as pd	
import datetime	
import pandas_datareader.data as web	
from pandas import Series, DataFrame	
start = datetime.datetime(2010, 1, 1)	
end = datetime.datetime(2017, 1, 11)	
df = web.DataReader("AAPL", 'yahoo', start, end)	
df.tail()

来源于雅虎财经的股票价格

这段代码将提取从 2010 年 1 月到 2017 年 1 月的 7 年的数据。你可以根据需要调整开始和结束日期。接下来的分析过程，我们将使用收盘价格，即股票在一天交易结束时的最终价格。

3 探索股票的移动平均值和收益率

在这个分析中，我们使用两个关键的测量指标来分析股票：移动平均值和回报率。

3.1 移动平均值：确定趋势

滚动平均 / 移动平均（MA）通过不断更新平均价格来平滑价格数据，有助于降低价格表中的“噪音”。此外，该移动平均线可能充当“阻力”，代表着股票的下跌和上升趋势，你可以从中预期它的未来趋势，不太可能偏离阻力点。

让我们开始写代码来得到滚动平均值:

1. close_px = df['Adj Close']

2. 
   

3. mavg = close_px.rolling(window=100).mean()

最后 10 个移动平均值

这将计算股票收盘价最后 100 个滑窗（100天）的移动平均值，并取每个滑窗的移动平均值。正如你所看到的，移动平均线在滑窗上稳步上升，并不遵循股票价格曲线的锯齿线。为了更好地理解，让我们用 Matplotlib 来绘制它。我们将用移动平均线来绘制股票价格表。

1. %matplotlib inline

2. import matplotlib.pyplot as plt

3. from matplotlib import style

4. 
   

5. # Adjusting the size of matplotlib

6. 
   

7. import matplotlib as mpl

8. mpl.rc('figure', figsize=(8, 7))

9. 
   

10. mpl.__version__

11. 
    

12. # Adjusting the style of matplotlib

13. 
    

14. style.use('ggplot')

15. 
    

16. close_px.plot(label='AAPL')

17. 
    

18. mavg.plot(label='mavg')

19. 
    

20. plt.legend()

苹果股票移动平均价格(mavg)

移动平均使曲线平滑，显示股票价格的涨跌趋势。

在这张图表中，移动平均线显示了股票价格上升或下降的趋势。从逻辑上讲，你应该在股市低迷时买进，在股市上涨时卖出。

3.2 回报偏差:用于确定风险和收益

预期收益衡量投资收益概率分布的平均值或预期值。投资组合的预期回报是通过将每项资产的权重乘以其预期回报，再加上每项投资的价值来计算的。（摘自投资百科，investopedia）

你可以参考以下公式：

收益公式

根据这个公式，我们可以画出收益，如下。

1. rets = close_px / close_px.shift(1) - 1

2. 
   

3. rets.plot(label='return')

收益率

从逻辑上讲，我们理想的股票收益应该尽可能高且稳定。如果你是风险规避者（像我一样），你可能希望避开这种股票，因为你看到 2013 年下跌了 10%。这个决定很大程度上取决于你对股票的总体看法和对其他竞争对手股票的分析。

4 分析竞争对手股票

在这部分中，我们将分析一家公司相对于其竞争对手的表现。假设我们对科技公司感兴趣，并想比较一下这些巨头：苹果（Apple）、通用电气（GE）、谷歌（Google）、IBM 和微软（Microsoft）。

1. dfcomp = web.DataReader(['AAPL', 'GE', 'GOOG', 'IBM', 'MSFT'],'yahoo',start=start,end=end)['Adj Close']

苹果、通用电气、谷歌、IBM 和微软的股价

你将会从雅虎财经的股票价格中得到一张相当整洁平滑的收盘价表。

4.1 相关性分析:竞争对手会互相影响吗？

我们可以通过运行 pandas 的百分比变化和相关函数来分析竞争关系。百分比变化将定义收益 ，找出与前一天相比价格变化的程度。了解相关性将有助于我们理解收益是否受其他股票收益的影响。

1. retscomp = dfcomp.pct_change()

2. 
   

3. corr = retscomp.corr()

让我们绘制散点图来观察 Apple 和 GE 的收益分布。

1. plt.scatter(retscomp.AAPL, retscomp.GE)

2. 
   

3. plt.xlabel('Returns AAPL')

4. 
   

5. plt.ylabel('Returns GE')

Apple 和 GE 的散点图

我们可以看到，GE 的收益和 Apple 的收益之间存在着微弱的正相关关系。似乎在大多数情况下， Apple 的回报率越高，GE 的回报率也就越高。

让我们通过绘制散点矩阵进一步改进我们的分析，以可视化竞争股票之间可能的相关性。在对角点，我们将运行核密度估计(Kernel Density Estimate,KDE)。KDE 是一个基本的数据平滑问题，它基于有限的数据样本对总体进行推断。它有助于生成对总体分布的估计。

1. pd.scatter_matrix(retscomp, diagonal='kde', figsize=(10, 10))

KDE 图和散点矩阵

从这里我们可以得到大多数股票之间的分布近似正相关。

为了证明正相关关系，我们将使用热图来可视化竞争股票之间的相关程度。注意颜色越浅代表这两只股票的相关性越强。

1. plt.imshow(corr, cmap='hot', interpolation='none')

2. plt.colorbar()

3. 
   

4. plt.xticks(range(len(corr)), corr.columns)

5. plt.yticks(range(len(corr)), corr.columns)

竞争股票之间相关性的热图

从散点矩阵和热图中我们可以发现，竞争股票之间有很大的相关性。然而，这可能并不能说明因果关系，只能说明科技行业的趋势而不能说明相互竞争的股票是如何相互影响的。

4.2 股票回报率和风险

除了相关性，我们还分析了每支股票的风险和回报。本例中我们提取的是回报的平均值（回报率）和回报的标准差（风险）。

1. plt.scatter(retscomp.mean(), retscomp.std())

2. plt.xlabel('Expected returns')

3. 
   

4. plt.ylabel('Risk')

5. 
   

6. for label, x, y in zip(retscomp.columns, retscomp.mean(), retscomp.std()):

7. 
   

8. plt.annotate(label,xy = (x, y), xytext = (20, -20),

9. textcoords = 'offset points', ha = 'right', va = 'bottom',

10. bbox = dict(boxstyle = 'round,pad=0.5', fc = 'yellow', alpha = 0.5),

11. arrowprops = dict(arrowstyle = '->', connectionstyle = 'arc3,rad=0'))

股票风险与收益的快速散点图

现在你可以看到这张关于竞争股票的风险和收益比较的清晰的图表。从逻辑上讲，你想要将风险最小化，并使收益最大化。因此，您需要为您的风险回报容忍度画一条线(红线)。然后，你将创建规则——购买红线以下的股票(微软、通用电气和 IBM)，卖出红线以上的股票(苹果和谷歌)。这条红线显示了您的期望值阈值和买进/卖出决策的基线。

5 预测股票价格

5.1 特征工程

我们将使用这三个机器学习模型来预测股票:简单线性分析、二次判别分析(Quadratic Discriminant Analysis，QDA)和 K近邻(K Nearest Neighbor，KNN)。但首先我们需要设计一些特征：高-低百分比和百分比变化。

1. dfreg = df.loc[:,['Adj Close','Volume']]

2. 
   

3. dfreg['HL_PCT'] = (df['High'] — df['Low']) / df['Close'] * 100.0

4. 
   

5. dfreg['PCT_change'] = (df['Close'] — df['Open']) / df['Open'] * 100.0

生成的最终数据帧

5.2 预处理和交叉验证

在将数据放入预测模型之前，我们将按照以下步骤对数据进行清洗和处理：

1.删除缺失值

2.分离标签，我们要预测 Adjclose

3.缩放 X ，使每个样本都可以具有相同的线性回归分布。

4.最后，我们要找到近期 X 和早期 X （用于训练）的数据序列，用于模型生成和评估。

5.分离标签并标识为 Y。

6.分别通过交叉验证训练模型和测试

请参考以下的代码。

1. # Drop missing value

2. dfreg.fillna(value=-99999, inplace=True)

3. 
   

4. # We want to separate 1 percent of the data to forecast

5. forecast_out = int(math.ceil(0.01 * len(dfreg)))

6. 
   

7. # Separating the label here, we want to predict the AdjClose

8. 
   

9. forecast_col = 'Adj Close'

10. dfreg['label'] = dfreg[forecast_col].shift(-forecast_out)

11. 
    

12. X = np.array(dfreg.drop(['label'], 1))

13. 
    

14. # Scale the X so that everyone can have the same distribution for linear regression

15. X = preprocessing.scale(X)

16. 
    

17. # Finally We want to find Data Series of late X and early X (train) for model generation and evaluation

18. X_lately = X[-forecast_out:]

19. X = X[:-forecast_out]

20. 
    

21. # Separate label and identify it as y

22. y = np.array(dfreg['label'])

23. y = y[:-forecast_out]

5.3 模型生成-预测过程有意思的地方开始了

首先，让我们为我们的 SciKit-Learn 库 的导入添加以下代码：

1. from sklearn.linear_model import LinearRegression

2. from sklearn.neighbors import KNeighborsRegressor

3. from sklearn.linear_model import Ridge

4. from sklearn.preprocessing import PolynomialFeatures

5. from sklearn.pipeline import make_pipeline

5.4 简单线性分析和二次判别分析

简单线性分析给出两个或多个变量之间的线性关系。当我们在两个变量中画关系图时，我们会得到一条直线。二次判别分析类似于简单线性分析，只是模型允许多项式(例如: x 平方)的生成并会产生曲线。

线性回归预测因变量(Y)为输出而自变量(X)为输入。在绘制过程中，我们会得到一条直线，如下图所示:

简单线性回归

以下是一篇相当干货的文章，它对线性回归进行了全面的回顾。

A beginner’s guide to Linear Regression in Python with Scikit-Learn https://towardsdatascience.com/a-beginners-guide-to-linear-regression-in-python-with-scikit-learn-83a8f7ae2b4f

我们将导入并使用现有的 SciKit 库，并通过选择 X 和 Y 训练集来训练模型。代码如下。

1. # Linear regression

2. clfreg = LinearRegression(n_jobs=-1)

3. clfreg.fit(X_train, y_train)

4. 
   

5. # Quadratic Regression 2

6. clfpoly2 = make_pipeline(PolynomialFeatures(2), Ridge())

7. clfpoly2.fit(X_train, y_train)

8. 
   

9. # Quadratic Regression 3

10. clfpoly3 = make_pipeline(PolynomialFeatures(3), Ridge())

11. clfpoly3.fit(X_train, y_train)

5.5 K最近邻(KNN)

KNN利用特征相似性来预测数据点的值。这保证了分配的新点与数据集中的点相似。为了找出相似点，我们提取这些点间的最小距离(例如：欧氏距离)。

在 KNN 模型可视化中，你将会以 k 个数量的元素对问题元素进行分组

有关模型的详细资料，请参阅以下链接。这对加深你的理解非常有用。

Introduction to k-Nearest Neighbors: Simplified (with implementation in Python) https://www.analyticsvidhya.com/blog/2018/03/introduction-k-neighbours-algorithm-clustering/

1. # KNN Regression

2. clfknn = KNeighborsRegressor(n_neighbors=2)

3. clfknn.fit(X_train, y_train)

5.6 评估

一种简单、快速、粗略的评估方法是在每个训练过的模型中使用 score 方法。score 可以得到测试数据集的 self.predict(X) 与 y 的平均精度。

1. confidencereg = clfreg.score(X_test, y_test)

2. confidencepoly2 = clfpoly2.score(X_test,y_test)

3. confidencepoly3 = clfpoly3.score(X_test,y_test)

4. confidenceknn = clfknn.score(X_test, y_test)

5. 
   

6. # results

7. 
   

8. ('The linear regression confidence is ', 0.96399641826551985)

9. 
   

10. ('The quadratic regression 2 confidence is ', 0.96492624557970319)

11. 
    

12. ('The quadratic regression 3 confidence is ', 0.9652082834532858)

13. 
    

14. ('The knn regression confidence is ', 0.92844658034790639)

结果显示大多数模型的精确性评分（>0.95）。然而，这并不意味着我们可以盲目地持有这些股票。仍然有许多问题需要考虑，特别是对于不同的公司来说，随着时间的推移，它们的价格走势也会不同。

为了进行全面的测试，让我们输出一些股票的预测。

1. forecast_set = clf.predict(X_lately)

2. 
   

3. dfreg['Forecast'] = np.nan

4. 
   

5. #result

6. 
   

7. (array([ 115.44941187, 115.20206522, 116.78688393, 116.70244946,

8. 
   

9. 116.58503739, 115.98769407, 116.54315699, 117.40012338,

10. 
    

11. 117.21473053, 116.57244657, 116.048717 , 116.26444966,

12. 
    

13. 115.78374093, 116.50647805, 117.92064806, 118.75581186,

14. 
    

15. 118.82688731, 119.51873699]), 0.96234891774075604, 18)

6 绘制预测图

根据预测值，我们将用现有的历史数据来可视化绘图。这将有助于我们可视化模型如何预测未来股票价格。

1. last_date = dfreg.iloc[-1].name

2. 
   

3. last_unix = last_date

4. 
   

5. next_unix = last_unix + datetime.timedelta(days=1)

6. 
   

7. for i in forecast_set:

8. 
   

9. next_date = next_unix

10. next_unix += datetime.timedelta(days=1)

11. dfreg.loc[next_date] = [np.nan for _ in range(len(dfreg.columns)-1)]+[i]

12. 
    

13. dfreg['Adj Close'].tail(500).plot()

14. dfreg['Forecast'].tail(500).plot()

15. plt.legend(loc=4)

16. plt.xlabel('Date')

17. plt.ylabel('Price')

18. plt.show()

图形表示预测值

如图所示，蓝线显示了基于回归的股票价格预测。预测表明，经济衰退不会持续太久，然后就会复苏。因此，我们可以在经济低迷时买进股票，在经济好转时卖出。

7 未来的改进/挑战

为了进一步分析股票，这里有一些你可以实现的想法。这些想法将有助于对股票进行更全面的分析。如果需要更多的说明，请随时通知我。

• 分析经济定性因素，如新闻（新闻来源和情感分析）

• 分析经济定量因素，如某个国家的HPI、公司起源之间的经济不平等

代码

在公众号『Python数据之道』后台回复 “code”，可以获取本文的代码文件

文章来源： https://towardsdatascience.com/in-12-minutes-stocks-analysis-with-pandas-and-scikit-learn-a8d8a7b50ee7

本文原作者 Vincent Tatan 的声明：

免责声明：本免责声明告知读者，本文中表达的观点、想法和意见仅属于作者本人，并不一定属于作者的雇主、组织、委员会或其他团体或个人。参考文献是从列表中挑选出来的，与其他作品的任何相似之处都纯属巧合。

本文写作纯粹是作者的副业，没有任何其他隐藏的动机。

本文来自公众号读者翻译，欢迎各位童鞋向公号投稿，点击下面图片了解详情！

译者简介

李洁，北京师范大学香港浸会大学联合学院 数据科学系助教，香港科技大学电信学硕士。

-------------------End-------------------

 Python数据之道 

据说学Python的只有10%的人关注了这个号，

还有很大潜力

今日主题：聊聊你对股票投资的一些见闻

留言格式：昵称 + day xx + 留言内容（字数不少于15字）

欢迎各位同学加入公众号读者分享交流群，在公众号后台回复 “微信群” 即可。

• 推荐 | 免费获取《Python知识手册》
  

• Matplotlib可视化最有价值的50个图表（附源码）

• 推荐 | 这才是你寻寻觅觅想要的Python可视化神器

• 推荐一个牛逼的生物信息 Python 库 - Dash Bio

同学们，支持就请右下角点！