テキストの数字列を読み込んでNumpy列に変換

MATLABに慣れるとNumpyの初期化周りがだるく感じる。
今回はtxtファイルに

4.0,8.1
2.2,1.1

みたいにして保存されている数字列を取り出してnumpyのarrayあたりで読み込めるようにする。

ソースコード

import numpy as np

#Read data
p_r, p_c = [], []
for l in open('HOGE.txt').readlines():
    # Data division
    data = l[:-1].split(',')
    # append or + 
    p_r.append([float(data[0]),float(data[1])])
    p_c.append([float(data[2]),float(data[3])])
# Data convert
p_r = np.array(p_r,dtype='float32')
p_c = np.array(p_c,dtype='float32')

解説メモ

splitのところでは,の代わりにスペース' 'やタブ'\t'などを指定しても良い。
appendのところは+=でも似たような挙動になるが多次元配列に変換したければappendが良さげ。
arrayの変換はnp.array(p_r,np.float32)でも可らしい。