首页 > 编程知识 正文

python中如何计算余弦距离公式,python中的余弦定理

时间:2023-05-06 19:33:46 阅读:249203 作者:1589

本篇文章小编给大家分享一下Python求向量的余弦值操作代码示例,文章代码介绍的很详细,小编觉得挺不错的,现在分享给大家供大家参考,有需要的小伙伴们可以来看看。

1、余弦相似度

余弦相似度衡量的是2个向量间的夹角大小,通过夹角的余弦值表示结果,因此2个向量的余弦相似度为:

余弦相似度的取值为[-1,1],值越大表示越相似。

向量夹角的余弦公式很简单,不在此赘述,直接上代码:

def cosVector(x,y):

if(len(x)!=len(y)):

print('error input,x and y is not in the same space')

return;

result1=0.0;

result2=0.0;

result3=0.0;

for i in range(len(x)):

result1+=x[i]*y[i] #sum(X*Y)

result2+=x[i]**2 #sum(X*X)

result3+=y[i]**2 #sum(Y*Y)

#print(result1)

#print(result2)

#print(result3)

print("result is "+str(result1/((result2*result3)**0.5))) #结果显示

cosVector([2,1],[1,1])

一个计算二维数组余弦值的例子:

#求余弦函数

def cosVector(x,y):

if(len(x)!=len(y)):

print('error input,x and y is not in the same space')

return;

result1=0.0;

result2=0.0;

result3=0.0;

for i in range(len(x)):

result1+=x[i]*y[i] #sum(X*Y)

result2+=x[i]**2 #sum(X*X)

result3+=y[i]**2 #sum(Y*Y)

#print("result is "+str(result1/((result2*result3)**0.5))) #结果显示

return result1/((result2*result3)**0.5)

#print("result is ",cosVector([2,1],[1,1]))

#计算query_output(60,20)和db_output(60,20)的余弦值,用60*1的向量存储

cosResult= [[0]*1 for i in range(60)]

for i in range(60):

cosResult[i][0]=cosVector(query_output[i], db_output[i])

print(cosResult)

--------------------------------------------------------------------------------------------

#计算query_output和db_output的余弦值,用60*1的向量存储

rows=query_output.shape[0] #行数

cols=query_output.shape[1] #列数

cosResult= [[0]*1 for i in range(rows)]

for i in range(rows):

cosResult[i][0]=cosVector(query_output[i], db_output[i])

#print(cosResult)

#将结果存入文件中,并且一行一个数字

file=open('cosResult.txt','w')

for i in cosResult:

file.write(str(i).replace('[','').replace(']','')+'n') #rn为换行符

file.close()

历史中提交的图片或压缩文件

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。