3.数据文件的读写

写入文件
我们先创建一个含有3阶单位矩阵的一个文件.numpy中创建单位矩阵有eye()函数来完成.

e=numpy.eye(3)

我们将这个矩阵保存在eye3.txt中

import numpy as np
e=np.eye(3)
print e
np.savetxt("eye3.txt",e)

读取文件
读取文件时,通常是读取数据库文件转储格式CSV.
例如股价存储数据:
AAPL,21-01-2011, ,344.17,344.4,333.53,336.1,2114480
第一列为股票识别码,第二列是日期,依次是开盘价,最高价,最低价,和收盘价,最后一列是成交量.
为了多出几列数据,我改了下数据,添加了共4列数据,返回了一个数组

numpy里的loadtxt可以读取CSV文件并且提出我们想要的

c,v=np.loadtxt('data.csv',delimiter=',',usecols=(6,7),unpack=true)

文件名称为data.csv,分割符是’,’,取第6,7两列数据,不打包即分开存储在c,v中.

In [1]: c,v=np.loadtxt('/home/syd/Documents/data.csv',delimiter=',',usecols=(6,7),unpack=True)

In [2]: c
Out[2]: array([ 100.,  200.,  300.,  400.])

In [3]: v
Out[3]: array([ 2114480.,  2114480.,  2114480.,  2114480.])