余弦相似度计算，使用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的余弦值