使用Python对股票数据进行数据分析(一）-计算日线行情、5日均线、10日均线行情并显示

全栈程序员-用户IM • 2022年10月26日下午6:16 • 未分类

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全家桶1年46，售后保障稳定

使用Python对股票数据进行数据分析(一）-计算日线行情、5日均线、10日均线行情并显示

各种炒股软件上可以显示各种技术指标，可以帮助投资者进行技术分析。这些股市中的这些指标都是怎么计算出来的呢？这里使用python的pandas库来进行计算。后期可能使用一些专门金融分析的库，比如talib库等进行分析。

一、获取数据

这里需要获取的数据是股票的日线行情，这里使用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']]

Jetbrains全家桶1年46，售后保障稳定

运行后结果如下：

使用Python对股票数据进行数据分析(一）-计算日线行情、5日均线、10日均线行情并显示

二、计算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)] = daily['close'].rolling(ma).mean()
print(daily)

输出结果，可以看到5日均线数据的前4天是没有数据的，因为这个时候还没有到5天，无法计算5日均线：

使用Python对股票数据进行数据分析(一）-计算日线行情、5日均线、10日均线行情并显示

三、使用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()

图片输出如下：

使用Python对股票数据进行数据分析(一）-计算日线行情、5日均线、10日均线行情并显示