Pyhton 中 CSV文件的存取

如下代碼在spyder的IPython中測試

1、np.savetxt（）將數組a中的元素保存爲csv格式 ：

np.savetxt(frame,array,fmt='%.18e',delimiter=None)

1  import numpy as np
2  a=np.arange(10).reshape(2,5)
4 a
6 array([[0, 1, 2, 3, 4],
7        [5, 6, 7, 8, 9]])
9 np.savetxt('a.csv',a,fmt='%d',delimiter=',')#fmt定義寫入文件的格式，例如：%d 以整數存儲；%.2f 以兩位浮點數存儲

2、np.loadtxt（）將CSV文件中的數據讀入到numpy中的一個數組中：

np.loadtxt(frame,dtype=np.float,delimiter=None,unpack=False)

1 b=np.loadtxt('a.csv',delimiter=',')
3 b5 array([[0., 1., 2., 3., 4.],
6        [5., 6., 7., 8., 9.]])

能夠觀察到上面生成的數組b的元素類型默認的是浮點數

能夠經過dtype設定生成的數組元素類型,以下：

1 c=np.loadtxt('a.csv',dtype=np.int,delimiter=',')
2 
3 c
4 Out[15]: 
5 array([[0, 1, 2, 3, 4],
6        [5, 6, 7, 8, 9]])

能夠觀察到上面生成的數組C的元素類型爲整數。

3、csv文件的侷限性：

csv文件只能有效存儲一維和二維數組。即：np.savetxt（）和  np.loadtxt（）只能有效存取一維和二維數組。

p.s:明天清明假期，準備認真體會一下春天的味道。