失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > python 移动平均线_使用python计算简单移动平均线

python 移动平均线_使用python计算简单移动平均线

时间:2024-02-19 18:55:48

相关推荐

python 移动平均线_使用python计算简单移动平均线

计算移动平均线是最常见的需求,实现起来也非常简单,下面来完成三件事:

1. 从csv格式的文件导入数据,数据内容如下图,注意数据列非常多,我们导入的时候只取了部分列:

2.计算简单算术移动平均线MA;

3.将计算好的数据输出到csv文件中。

我这里直接把代码写出来,复制下来就能运行,文件路径还有个别参数需要修改:

------------------------------------------------------------------------------------------------------

# -*- coding: utf-8 -*-

"""

@author: Michael Liang

@date:1020

"""

importpandasaspd

# ========== 从原始csv文件中导入股票数据,以特锐德sz300001为例

# 导入数据 - 注意:这里请填写数据文件在您电脑中的路径,因为原始数据列太多,所以只取了部分列

input_data = pd.read_csv('C:\\Users\\Mic\\PycharmProjects\\test\\data\\sz300001.csv',

usecols=['股票代码','股票名称','交易日期','开盘价','最高价','最低价','收盘价','后复权价','前复权价'],

parse_dates=['交易日期'])

pd.set_option('expand_frame_repr', False)

# 将数据按照交易日期从远到近排序

input_data = input_data.sort_values(by='交易日期',ascending=1)

# ========== 计算移动平均线(以‘前复权价’为例)

# 分别计算5日、20日、60日的移动平均线

malist = [5,20,60]

# 计算简单算术移动平均线MA - 注意:stock_data['close']为股票每天的收盘价

formainmalist:

input_data['MA_'+str(ma)] = input_data['前复权价'].rolling(ma).mean()

# ========== 将算好的数据输出到csv文件 - 注意:这里请填写输出文件在您电脑中的路径,这里用gbk编码方便用Excel打开展示,实际csv建议输出utf-8格式

input_data.to_csv('C:\\Users\\Mic\\PycharmProjects\\test\\data\\sz300001_ma.csv',encoding='gbk',index=False)

---------------------------------------------------------------------------------------------------------------

输出的数据截图如下:

(完)

如果觉得《python 移动平均线_使用python计算简单移动平均线》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。