配列の要素に小数を代入しても整数になってしまいます。

Question

配列の要素に小数を代入しても整数になってしまいます。
小数を出力する方法を教えていただきたいです。

よろしくお願い致します。

import numpy as np

X = np.array([[0,0],[0,0]]);
X[0,0] = 2.156789

print(X[0,0]);

2

Answer 1

numpy 配列の型は dtype 引数で指定しない場合は、第1引数の値で推定されます。
0 は int 型なので、各値を 0. とするか、dtype=float と指定します。

X = np.array([[0, 0],
              [0, 0]])
print(X.dtype)  # int64

X = np.array([[0., 0.],
              [0., 0.]])
print(X.dtype)  # float64

import numpy as np

X = np.array([[0, 0],
              [0, 0]], dtype=float)
print(X.dtype)  # float64

X[0, 0] = 2.156789
print(X[0, 0])  # 2.156789

Comments

[mizumimizu]

わかりやすいご回答ありがとうございました。
おかげさまで理解することができました。