换个姿势学量化!|【量化小讲堂】使用python计算各类移动平均线

作者:邢不行

原文链接:

http://bbs.pinggu.org/thread-3631776-1-1.html

(本文已获作者授权转载,如需转载请与原作者联系)


【量化小讲堂-python & pandas技巧系列】使用python计算各类移动平均线

计算移动平均线是最常见的需求,下面这段代码将完成以下三件事情: 1. 从csv格式的文件中导入股票数据,数据例图如下:

2.计算各类移动平均线,包括简单简单算术移动平均线MA、指数平滑移动平均线EMA; 3.将计算好的数据输出到csv文件中。 代码应该复制下来就能运行了,关于从哪里可以得到代码中使用的数据,后面会讲,下面贴上代码: # -*- coding: utf-8 -*- """ @author: yucezhe @contact: QQ:2089973054 email:xjc@yucezhe.com """ import pandas as pd # ========== 从原始csv文件中导入股票数据,以浦发银行sh600000为例 # 导入数据 - 注意:这里请填写数据文件在您电脑中的路径 stock_data = pd.read_csv('stock data/sh600000.csv', parse_dates=[1]) # 将数据按照交易日期从远到近排序 stock_data.sort('date', inplace=True) # ========== 计算移动平均线 # 分别计算5日、20日、60日的移动平均线 ma_list = [5, 20, 60] # 计算简单算术移动平均线MA - 注意:stock_data['close']为股票每天的收盘价 for ma in ma_list: stock_data['MA_' + str(ma)] = pd.rolling_mean(stock_data['close'], ma) # 计算指数平滑移动平均线EMA for ma in ma_list: stock_data['EMA_' + str(ma)] = pd.ewma(stock_data['close'], span=ma) # 将数据按照交易日期从近到远排序 stock_data.sort('date', ascending=False, inplace=True) # ========== 将算好的数据输出到csv文件 - 注意:这里请填写输出文件在您电脑中的路径 stock_data.to_csv('sh600000_ma_ema.csv', index=False) 代码输出的数据截图如下:

可以到这里下载程序中用到的股票数据:(http://yucezhe.com/product?name=trading-data),其中包含了所有股票、从上市日起的全部交易数据。

原文发布于微信公众号 - 数说工作室(shushuojun)

原文发表时间:2015-11-26

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏深度学习与数据挖掘实战

干货|PyTorch实用代码段集锦

Facebook公司开源的深度学习框架PyTorch越来越火,PyTorch易于上手。本文节选github中PyTorch的常用实用代码片段,供大...

2927
来自专栏深度学习之tensorflow实战篇

Python信贷数据处理与初步分析(ZIP解压)

#!/usr/bin/python # coding=utf-8 ''' @author: lenovo @software: 3.6 PyChar...

4067
来自专栏Spark学习技巧

第3篇:更新异常与规范化设计

第三篇:更新异常与规范化设计 前言 在前两篇中,主要讲了ER建模和关系建模。在具体分析如何用数据库管理软件RDBMS(Relational Database M...

4357
来自专栏青玉伏案

算法与数据结构(八) AOV网的关键路径(Swift版)

上篇博客我们介绍了AOV网的拓扑序列,请参考《数据结构(七) AOV网的拓扑排序(Swift面向对象版)》。拓扑序列中包括项目的每个结点,沿着拓扑序列将项目进行...

2288
来自专栏天天P图攻城狮

OpenGL shader性能优化策略(一):减少分支语句

一、优化策略:减少使用分支语句 在编写OpenGL shader时,一定要注意减少使用if或for语句,因为这些语句引入分支、会大大降低shader的性能,...

6212
来自专栏大数据挖掘DT机器学习

《小美好》短评文本情感分析+生成词云

因为最近看了一下《致我们单纯的小美好》,虽然情节是有点“二”吧,但是看了觉得真的很怀念初高中的日子,一时玩心大发,于是就想搞点有意思的东西。。。首先去爬了豆瓣...

4199
来自专栏窗户

scratch写的图灵机

1473
来自专栏大数据文摘

干货 | 男朋友老是说自己R语言很6,快来用这40道题目检测他

2044
来自专栏奇点大数据

Github寻宝 | 贪吃蛇游戏AI版,代码就得这么写!

本期案例是一个C++ 项目,同时也是经典小游戏——贪吃蛇的升级版。(该项目由Github用户stevennl贡献,英文原版可访问Github网站:https:/...

3814
来自专栏PPV课数据科学社区

【教程】一小时向非程序员介绍 R 编程语言

(1)下载R和RStudio 我对RStudio的印象不错,对于初学者来说,它既方便又很有帮助,对专业人士也很有用。尤其对于初学者:鼠标指向-点击式(point...

2988

扫码关注云+社区

领取腾讯云代金券