使用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天是没有数据的，因为这个时候还没有到5天，无法计算5日均线：

三、使用matplotlib对数据进行作图

使用matplotlib进行作图，以日期列作为横坐标，收盘价、5日均线、10日均线、20日均线作为纵坐标进行作图显示，并保存图像。

#收盘价作图
plt.plot(daily['trade_date'], daily['close'], '-', label = 'close')
 
for ma in ma_list:
    #ma作图
    plt.plot(daily['trade_date'], daily['ma_' + str(ma)], '-', label = 'ma_' + str(ma))
 
plt.gcf().autofmt_xdate()
#显示文字
plt.legend()
#显示图片
plt.savefig('all.png')
plt.show()
图片输出如下：

使用tushare pro 数据需要免费注册，取得token。点击免费注册获取token或者https://tushare.pro/register?reg=287337，免费注册，获取token。

就到这里了，看看是不是和股票软件显示的一样的。有个这个数据是不是可以激发你的灵感。比如自己写一个程序，当10日均线小于5日均线，并且10日均线大约20日均线的时候，发出提醒。赶紧动起手来吧，实现你的千奇百怪的想法，祝各位早日实现财务自由的梦想~~