《Numpy学习指南》练习6-分析CSV中数据(附下载)

data.csv下载链接

链接:http://pan.baidu.com/s/1dFkyybv 密码:xh86

loadtxt 读入csv

首先读入一个股票的交易记录。csv中数据如下

《Numpy学习指南》练习6-分析CSV中数据(附下载)_第1张图片
屏幕快照 2017-10-05 下午2.14.47.png

从A到H列分别为
公司名[0],日期[1],空[2],开盘价[3],最高价[4],最低价[5],收盘价[6],交易量[7]
读入csv后,其对应的索引是从0到7的。

我们现在先算下该公司的交易额。

通过loadtxt读入其收盘价和交易量

《Numpy学习指南》练习6-分析CSV中数据(附下载)_第2张图片
屏幕快照 2017-10-05 下午2.21.04.png

其中,

  1. 'data.csv'是文件名,当你的文件不在你所编写脚本的文件夹下时,这里就还要带上文件路径。
  2. delimiter指的是分割符,在csv中是以英文逗号来进行分割的
  3. usecols指的是按索引取出csv文件中哪几列
  4. unpack=True表示分拆储存不同的列的数据,即分别将收盘价和成交量赋值给c和v。

mean 计算收盘价算数平均值

通过mean方法,可以直接计算收盘价的算数平均值

屏幕快照 2017-10-05 下午2.30.11.png

average() 计算加权平均值

VWAP(成交量加权平均价格)表示金融资产的平均价格。某个价格成交量越高,该价格所占权重就越大。
通过average函数,设置weights变量为交易量,即可计算VWAP

屏幕快照 2017-10-05 下午2.36.27.png

最大值与最小值

通过max和min方法即可实现

屏幕快照 2017-10-05 下午2.46.03.png

计算极差

即某列数据最大值与最小值的差值,通过ptp方法实现

《Numpy学习指南》练习6-分析CSV中数据(附下载)_第3张图片
屏幕快照 2017-10-05 下午2.48.18.png

median 中位数

《Numpy学习指南》练习6-分析CSV中数据(附下载)_第4张图片
屏幕快照 2017-10-05 下午2.50.31.png

var 方差

《Numpy学习指南》练习6-分析CSV中数据(附下载)_第5张图片
屏幕快照 2017-10-05 下午2.52.24.png

计算股票收益率

当日股票收益率的公式为(今日股价-昨日股价)/昨日股价,其实就是环比嘛……

  1. 计算差值
    首先,通过diff函数,就可以计算出每日收盘价与昨日收盘价的差值
《Numpy学习指南》练习6-分析CSV中数据(附下载)_第6张图片
屏幕快照 2017-10-05 下午3.12.45.png
  1. 计算收益率
    然后,用昨日的收盘价做分母,今日和昨日收盘价当分子,就计算出收益率了
《Numpy学习指南》练习6-分析CSV中数据(附下载)_第7张图片
屏幕快照 2017-10-05 下午3.15.31.png

其中,c[:-1]表示从头取值,一直取到倒数第一个(-1),但是不包含最后一个

  1. 查看收益率>0的索引
    通过where函数,可以找出收益率大于0的值的对应索引。单独看索引没有意义,但配合日期等数据,就能很方便的找出符合条件的日期,计算时间分布等。
    我们先将收益率赋值给变量returns,然后在where的参数里直接写returns>0,即可返回returns>0的索引,与上面的retruns是一一对应的。


    《Numpy学习指南》练习6-分析CSV中数据(附下载)_第8张图片
    屏幕快照 2017-10-05 下午3.23.37.png

你可能感兴趣的:(《Numpy学习指南》练习6-分析CSV中数据(附下载))