twitter facebook

PythonのnumpyでCSVファイルの読み書き。

以下のCSVファイルが /temp/test.csv にあるとします。
/temp/test.csv
# cat /temp/test.csv
10,11,12,13 20,21,22,23

CSVデータの読み込み

python
import numpy as np
a = np.loadtxt('/temp/test.csv', delimiter=',')
print(a)
結果
array([[ 10.,  11.,  12.,  13.],
       [ 20.,  21.,  22.,  23.]])

CSVデータの操作

読み込み後はNumpyで操作できます。
b = a+100

print(b)
array([[ 110.,  111.,  112.,  113.],
       [ 120.,  121.,  122.,  123.]])

CSVデータの保存

np.savetxt('/temp/test2.csv', b, delimiter=',', fmt='%d')
CSVファイルが作成されているか確認。
# cat /temp/test2.csv 110,111,112,113 120,121,122,123

その他

ちょっとした計算は numpy で扱えますが、より複雑なデーター処理をする場合は scipy のほうが向いています。
AUTHOR
@hiropon
最終更新日 2018/09/18
FAVORITE good stock
LINK TAG
記法を見る