ハムスターに飼われる院生のブログ

自分用メモが中心のブログです。

pandasでcsvファイルを読み込んでnumpyのarrayに入れる

pythoncsvファイルを読み込んでnparrayに入れるためのサンプルコードを以下に述べる。なお、python2.7を用いているので、python3系とは異なる部分があるかもしれない。

 

import pandas as pd
import numpy as np

data= pd.read_csv('readme.csv',sep=",")#typeに注意。np.arrayとは異なるのでこの後変換する

#np.arrayに変換
npdata=np.array(data.values.flatten())

#npdataの形を、pandaで読み込んだデータフレームの形に変形する
nparray=np.reshape(npdata,(data.shape[0],data.shape[1]))

 

これで、ファイルreadme.csvファイルの中身を、numpyの行列形式に格納できた。

行列計算を行いたいデータを読み込む時によく使用している。