copipe Python | numpyの小数点以下桁数を指定する。

概要

printの前にnp.set_printoptions(precision=桁数)を使うことで、numpy配列の小数点以下桁数を指定することができる。

コード

import numpy as np
x = np.arange(1,10,1)
y = np.cos(x)
np.set_printoptions(precision=2)
print(y)

結果

[ 0.54 -0.42 -0.99 -0.65  0.28  0.96  0.75 -0.15 -0.91]