问题导读:
1.什么是pandas-datareader? 2.怎样使用DataReader分析股票数据? 3.怎样可视化股价数据?
pandas-datareader介绍
Pandas库提供了专门从财经网站获取金融数据的API接口,可作为量化交易股票数据获取的另一种途径,该接口在urllib3库基础上实现了以客户端身份访问网站的股票数据。需要注意的是目前模块已经迁徙到pandas-datareader包中,因此导入模块时需要由import pandas.io.data as web更改为import pandas_datareader.data as web。
相关参考网站
官网
github
0.6.0版本操作文档
DataReader方法介绍
查看Pandas的操作文档可以发现,第一个参数为股票代码,苹果公司的代码为"AAPL",国内股市采用的输入方式“股票代码”+“对应股市”,上证股票在股票代码后面加上“.SS”,深圳股票在股票代码后面加上“.SZ”。DataReader可从多个金融网站获取到股票数据,如“Yahoo! Finance” 、“Google Finance”等,这里以Yahoo为例。第三、四个参数为股票数据的起始时间断。返回的数据格式为DataFrame。
[mw_shl_code=python,true]start = datetime.datetime(2017,1,1)#获取数据的时间段-起始时间
end = datetime.date.today()#获取数据的时间段-结束时间
stock = web.DataReader("600797.SS", "yahoo", start, end)#获取浙大网新2017年1月1日至今的股票数据
[/mw_shl_code]
数据分析
1、打印DataFrame数据前5行和尾部倒数5行(浙大网新因重大事件停牌至1月6日开始复牌)
[mw_shl_code=python,true]print stock.head(5)
print stock.tail(5)
[/mw_shl_code]
2、打印DataFrame数据索引和列名,索引为时间序列,列信息为开盘价、最高价、最低价、收盘价、复权收盘价、成交量
[mw_shl_code=python,true]print stock.index
print stock.columns[/mw_shl_code]
3、打印DataFrame数据形状,index长度为248,columns数为6,即248个交易日,6项股票数据
[mw_shl_code=python,true]print stock.shape
[/mw_shl_code]
4、打印DataFrame数据查看数据是否有缺失,以及每列数据的类型
[mw_shl_code=python,true]print stock.info()
[/mw_shl_code]
5、打印DataFrame数据每组的统计情况,如最小值、最大值、均值、标准差等
[mw_shl_code=python,true]print stock.describe()[/mw_shl_code]
6、DataFrame数据中增加涨/跌幅列,涨/跌=(当日Close-上一日Close)/上一日Close*100%
(1)添加一列change,存储当日股票价格与前一日收盘价格相比的涨跌数值,即当日Close价格与上一日Close的差值,1月3日这天无上一日数据,因此出现缺失
[mw_shl_code=python,true]change = stock.Close.diff()
stock['Change'] = change
print stock.head(5)[/mw_shl_code]
(2)对缺失的数据用涨跌值的均值就地替代NaN。
[mw_shl_code=python,true]change.fillna(change.mean(),inplace=True)[/mw_shl_code]
(3)计算涨跌幅度有两种方法,pct_change()算法的思想即是第二项开始向前做减法后再除以第一项,计算得到涨跌幅序列。
[mw_shl_code=python,true]stock['pct_change'] = (stock['Change'] /stock['Close'].shift(1))#
stock['pct_change1'] = stock.Close.pct_change()
[/mw_shl_code]
7、DataFrame数据中增加跳空缺口数值序列,这里定义的缺口为上涨趋势和下跌趋势中的突破缺口,上涨趋势中今天的最低价高于昨天收盘价为向上跳空,下跌趋势中昨天收盘价高于今天最高价为向下跳空。遍历每个交易日后将符合跳空缺口条件的交易日增加缺口数值。
[mw_shl_code=python,true]jump_pd = pd.DataFrame()
for kl_index in np.arange(1, stock.shape[0]):
today = stock.ix[kl_index]
yesday = stock.ix[kl_index-1]
today['preCloae'] = yesday.Close
if today['pct_change'] > 0 and (today.Low-today['preCloae']) > 0:
today['jump_power'] = (today.Low-today['preCloae'])
elif today['pct_change'] < 0 and (today.High-today['preCloae']) < 0:
today['jump_power'] = (today.High-today['preCloae'])
jump_pd = jump_pd.append(today)
stock['jump_power'] = jump_pd['jump_power']
print stock.loc["2017-04-26":"2017-06-15"]#默认打印全部列
[/mw_shl_code]
8、DataFrame数据保留两位小数显示
[mw_shl_code=python,true]format = lambda x: '%.2f' % x
stock = stock.applymap(format)
print stock.loc["2017-04-26":"2017-06-15"]#默认打印全部列[/mw_shl_code]
股价数据的可视化Matplotlib是使用Python进行绘图里非常方便的库。这次 plot使用的数据是 Adj Close栏的数据。这是所说的已调整收盘价。 如下仅仅需要两行写就可以简单的将股价作为时间序列数据画出来。
[mw_shl_code=python,true]alibaba['Adj Close'].plot(legend=True, figsize=(10,4))
plt.show()
[/mw_shl_code]
完整示例
[mw_shl_code=python,true]import pandas as pd
from pandas import Series, DataFrame
import numpy as np
import matplotlib.pyplot as plt
from pandas_datareader import data, wb
from datetime import datetime
end = datetime.now()
start = datetime(end.year - 1, end.month, end.day)
alibaba = data.DataReader('BABA', 'yahoo', start, end)
alibaba['Adj Close'].plot(legend=True, figsize=(10,4))
plt.show()
[/mw_shl_code]
更多图表操作可参考:
实例操作:Python提取雅虎财经数据,并做数据分析和可视化
以csv格式存放
[mw_shl_code=python,true]import numpy as np
import pandas as pd
import pandas_datareader.data as web
import datetime
df_csvsave = web.DataReader("600018.SS","yahoo",datetime.datetime(2019,1,1),datetime.date.today())
print (df_csvsave)
df_csvsave.to_csv(r'C:\Users\15461\Desktop\table.csv',columns=df_csvsave.columns,index=True)
[/mw_shl_code]
可能遇到的问题
在《利用Python进行数据分析》一书中,学习到pandas这一章节,利用书中的方法去获取雅虎数据会出现以下问题:
问题一 The pandas.io.data module is moved to a separate package (pandas-datareader)
[mw_shl_code=python,true]import pandas.io.data as web
Traceback (most recent call last):
File "<pyshell#0>", line 1, in <module>
import pandas.io.data as web
File "C:\Program Files (x86)\Python36-32\lib\site-packages\pandas\io\data.py", line 2, in <module>
"The pandas.io.data module is moved to a separate package "
ImportError: The pandas.io.data module is moved to a separate package (pandas-datareader).
After installing the pandas-datareader package ( https://github.com/pydata/pandas-datareader),
you can change the import ``from pandas.io import data, wb`` to ``from pandas_datareader import data, wb``.
[/mw_shl_code]
根据提示,我们很快可以知道,‘ pandas.io’这个模块已经变成了‘pandas_datareader’了,cmd命令行中执行以下语句即可解决这个问题:
[mw_shl_code=shell,true]pip install pandas_datareader[/mw_shl_code]
用以下语句可以调用模块
[mw_shl_code=python,true]import pandas_datareader.data as web[/mw_shl_code]
问题二 Yahoo Actions has been immediately deprecated due to large breaks
[mw_shl_code=python,true]web.get_data_yahoo('AAPL','1/1/2000','1/1/2010')
Traceback (most recent call last):
File "<pyshell#2>", line 1, in <module>
web.get_data_yahoo('AAPL','1/1/2000','1/1/2010')
File "C:\Program Files (x86)\Python36-32\lib\site-packages\pandas_datareader\data.py", line 63, in get_data_yahoo
raise ImmediateDeprecationError(DEP_ERROR_MSG.format('Yahoo Actions'))
pandas_datareader.exceptions.ImmediateDeprecationError:
Yahoo Actions has been immediately deprecated due to large breaks in the API without the
introduction of a stable replacement. Pull Requests to re-enable these data connectors are welcome.
See https://github.com/pydata/pandas-datareader/issues
[/mw_shl_code]
原因 yahoo的数据接口有改动导致不可用。
在网上搜了好久才找到这一方法,这里我们需要引入另外一个模块‘fix_yahoo_finance’,同样使用pip方法进行安装
[mw_shl_code=shell,true]pip install fix_yahoo_finance[/mw_shl_code]
调用方式如下:
[mw_shl_code=python,true]import pandas_datareader.data as web
import datetime
import fix_yahoo_finance as yf
yf.pdr_override()
start=datetime.datetime(2006, 10, 1)
end=datetime.datetime(2012, 1, 1)
web.get_data_yahoo('AAPL',start,end)
[/mw_shl_code]
这样子就可以下载雅虎数据了。
问题三-在键入1/1/2000和1/1/2010后报错这里提示是时间的格式错误,应改为"%Y-%m-%d"
解决办法:for循环这里这样写
[mw_shl_code=python,true]for ticker in ['AAPL','IBM','MSFT','GOOG']:
...: all_data[ticker]=web.get_data_yahoo(ticker,'2000-1-1','2010-1-1')[/mw_shl_code]
来源: csdn 作者: 张小凡
原文链接:python积累--pandas---使用pandas-datareader获取股票信息进行分析
|