データ分析エンジニアのブログ

日常のことからプログラミングや機械学習まで@六本木

Python/Numpyでベクトルのユークリッド距離を求める


PythonのNumpyを使えば、次元によらずに
ベクトル同士のユークリッド距離が簡単に求められます。

方法1

>>> import numpy as np
>>> np.linalg.norm(a-b)

方法2

>>> import numpy as np
>>> np.sqrt(numpy.power(a-b, 2).sum())

おまけでScipyを使ってコサイン類似度を求める方法

>>> # コサイン類似度
>>> import scipy.spatial.distance as dis
>>> dis.cosine(a, b)