python cosine余弦相似计算

根据网上文章进行整理记录。加上自己的过程的理解,加深印象

# 假设三维空间,矢量信息如下
a = [1,2,3]
b = [4,5,6]

""" 
Cosine相似计算公式

a = [X₁,Y₁,Z₁]
b = [X₂,y₂,Z₂]

            X₁*X₂ + Y₁*y₂ + Z₁*Z₂
Cos(α) = -----------------------------------------------
         __________     __________
        √ X₁² + Y₁² + Z₁²  * √ X₂² + y₂² + Z₂²
       
"""
# 分子矢量相乘累加
sum_xy = 0.0

# X₁² + Y₁² + Z₁²
normA = 0.0

# X₂² + y₂² + Z₂²
normB = 0.0

# 分别遍历a和b => (X₁,X₂) (Y₁,y₂)  (Z₁,Z₂)
for X₁, X₂ in zip(a,b):

    # 矢量相乘累加
    sum_xy += X₁*X₂
    
    # 计算|A|部分。自己的矢量平方相加
    normA += X₁ ** 2

    # 计算|B|部分。自己的矢量平方相加
    normB += X₂ ** 2

# 开根号。 即可2分支1的平方, 即  ** 0.5
# normA和normB根号相乘,可以先相乘再开根号
print(sum_xy / ((normA * normB) ** 0.5))

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值