Numpy:回転行列がうまくいかない

Question

前提・実現したいこと

二次元の座標を回転させたい

発生している問題・エラーメッセージ

座標[0, 1]を90度回転させたいが、結果がおかしい

該当のソースコード

import numpy as np

R = lambda rad: np.array([[np.cos(rad), -np.sin(rad)],
                          [np.sin(rad), np.cos(rad)]])
print(R(np.radians(90))@[0, 1])
#array([-1.000000e+00,  6.123234e-17])

補足情報（FW/ツールのバージョンなど）

Python 3.9.2
Numpy 1.20.2
Windows 10

Answer 1

[0, 1]から90度回転で[-1, 0]になるので、「ほぼ」合っています。[-1.000000e+00, 6.123234e-17]という形で出てくるのは、浮動小数点数で三角関数を計算する都合上、誤差があるからです。
（指数表記に馴染みがないかもしれませんが、e±x部分が*10^xを意味します。なので、得られている値は「ほぼ」[-1, 0]であり、誤差が表示されているだけです。）

邪魔なら結果を適当な精度で丸める手があります。
numpy.ndarray.round — NumPy v1.20 Manual

Comments

[sakuramochi_py]

ありがとうございます！
試しにnp.roundで丸めてみたら正しい値が導きだせました！