使用Python对股票数据进行数据分析(二）-使用ta-lib库获取日线行情、5日均线、10日均线行情并显示

使用Python对股票数据进行数据分析(二）-使用ta-lib库获取日线行情、5日均线、10日均线行情并显示

这个文章其实和我的上一篇文章，https://blog.csdn.net/lost0910/article/details/104572950有点类似，只是数据处理部分有点不一样。

一、获取数据

这里需要获取的数据是股票的日线行情，这里使用tushare进行获取，以‘贵州茅台’股票进行分析。

#获取贵州茅台2019年12月1日以来的日线数据
daily = pro.daily(ts_code = '600519.SH',start_date = '20191201')
#将trade_date列，转化为日期格式
daily['trade_date'] = pd.to_datetime(daily['trade_date'], format="%Y%m%d")
#删除无用的列
daily = daily[['trade_date', 'close']]
运行后结果如下：

二、计算5日均线、10日均线数据

代码如下，通过pandas的rolling函数，计算5日均线数据：

#按照时间顺序升序排列
daily.sort_values('trade_date', inplace = True)
ma_list = [5, 10, 20]
for ma in ma_list:
    daily['ma_' + str(ma)] = talib.MA(daily['close'], timeperiod=ma)
print(daily)
输出结果，可以看到5日均线数据的前4天是没有数据