餘弦相似度計算，使用scipy.spatial.distance

引入模塊

from scipy.spatial.distance import pdist

pdlist函數使用

a=[1,1,1]
b=[1,1,2]
c=[1,1,2]
d=[1,1,2]
e=[1,1,2]

# cos
print("cos:",pdist([a,b,c,d,e],"cosine"))

1. 第一個參數傳入一個矩陣

2. 第二個參數傳入需要計算的類型

   • cosine 餘弦值
   • euclidean 歐式距離

3. 返回值爲一個列表[ab,ac,ad,ae,bc,bd,be,cd,ce,de],ab代表a和b的餘弦值