如何在python中使用廣播計算矩陣的成對余弦距離

Question

我有一個這樣的矩陣，其中每一行都是一個向量，需要在不使用循環的情況下計算向量之間的成對余弦距離（在本例中，作為 2x2 矩陣）？ 我知道笛卡爾距離可以填充一個維度，但不知道這里的最佳方法。

m = np.array([[1, 3, 5],[2,6,10]])

Answer 1

這種類型的問題會讓你斥責“堆棧不適合做作業”，但我想你可以這樣做......

from numpy import linalg as LA
import numpy as np
m = np.array([[1, 3, 5],[2,6,10]])
cos = float(sum(m[0]*m[1]))/float(LA.norm(m[0])*LA.norm(m[1]))

當然，如果你想避免使用linlag

float(sum(m[0]*m[1]))/(float(sum(m[0]*m[0]))**0.5*float(sum(m[1]*m[1]))**0.5)