2012-03-30 6 views
9

これに近い回答が多数ありますが、私はそれらを理解していないか、質問に答えていません。numpy再配列のdtypeを変更するにはどうすればよいですか?

numpy.rec.fromrecordsを使って再作成しました。特定の列を浮動小数点数に変換したいとします。これはどうすればいいですか?私はndarrayに変更し、再配列に戻すべきですか?ここ

答えて

14

は、変換を実行するastypeを用いた例である。

import numpy as np 
recs = [('Bill', '31', 260.0), ('Fred', 15, '145.0')] 
r = np.rec.fromrecords(recs, formats = 'S30,i2,f4', names = 'name, age, weight') 
print(r) 
# [('Bill', 31, 260.0) ('Fred', 15, 145.0)] 

ageはDTYPE <i2である:

print(r.dtype) 
# [('name', '|S30'), ('age', '<i2'), ('weight', '<f4')] 

我々はastypeを用い<f4にそれを変更することができる:

r = r.astype([('name', '|S30'), ('age', '<f4'), ('weight', '<f4')]) 
print(r) 
# [('Bill', 31.0, 260.0) ('Fred', 15.0, 145.0)] 
+0

ありがとう! "astype"は、新しい配列を作り直すよりもややコンパクトです...私はそれが効率の点で同じものになると仮定しています。私は既存のdtypeを変更する方法を含んでいるので、以下の私のソリューションも掲載しています。 – mathtick

11

基本的に2つのステップ。私の足かせは、既存のdtypeを変更する方法を見つけることでした。これが私のやり方です:

# change dtype by making a whole new array 
dt = data.dtype 
dt = dt.descr # this is now a modifiable list, can't modify numpy.dtype 
# change the type of the first col: 
dt[0] = (dt[0][0], 'float64') 
dt = numpy.dtype(dt) 
# data = numpy.array(data, dtype=dt) # option 1 
data = data.astype(dt) 
関連する問題