深入浅出Python量化交易实战
上QQ阅读APP看书,第一时间看更新

3.3.1 获取股票数据

首先我们使用之前学过的datareader来获取股票数据,这里需要导入一些必要的库,输入代码如下:

#导入Pandas
import pandas as pd
#导入金融数据获取模块datareader
import pandas_datareader.data as web
#导入numpy,一会儿会用到
import numpy as np

运行代码,如果程序没有报错,就说明导入成功。接下来,我们可以定义一个获取股票数据的函数,以便未来还可以复用。输入代码如下:

运行代码之后,就完成了函数的定义。下面就可以使用这个函数来获取数据。输入代码如下:

运行代码,会得到以下结果:

文件未找到,重新下载中
下载完成

【结果分析】因为这里是第一次使用load_stock函数来获取数据,所以程序会提示没有找到文件,并重新开始下载文件。稍等片刻之后,我们便可以看到程序告知数据下载完成。

如果读者朋友想要查看已经下载的数据,则可以使用下面这行代码:

#查看数据的前五行
zgpa.head()

运行代码,可以得到如表3.1所示的结果。

表3.1 获取的交易数据的前5行

【结果分析】从表3.1中可以看到,股票数据已经成功加载,包括的字段有Date(日期)、High(最高价)、Low(最低价)、Open(开盘价)、Close(收盘价)、Volume(成交量),和Adj Close(调整后的收盘价)。