首页 > 试题广场 >

计算向量之间的余弦相似度

[编程题]计算向量之间的余弦相似度

实现一个函数来计算两个向量之间的余弦相似度。余弦相似度是通过测量两个向量之间的夹角的余弦值来度量它们之间的相似性。

第一行为向量 ，可以是一维或二维数组
第二行为向量 ，维度需要与  相同

输出一个浮点数，表示两个向量之间的余弦相似度，结果保留3位小数。

示例1

输入

[1, 2, 3]
[4, 5, 6]

0.975

1.Python3对应的输入、输出已给出，您只用实现核心功能函数即可。
2.支持numpy、scipy、pandas、scikit-learn库。

算法知识视频讲解

Python3

马保深

def cosine_similarity(v1, v2):

# Implement your code here

v1=np.array(v1)

v2=np.array(v2)

dot=v1.dot(v2)

norm_v1=np.linalg.norm(v1,ord=2)

norm_v2=np.linalg.norm(v2,ord=2)

norm=norm_v1 * norm_v2

val=dot/norm

return round(val, 3)

发表于 2026-04-19 15:47:50 回复(0)

提交观点

线性代数

难度：

1条回答 4收藏 181浏览

计算向量之间的余弦相似度

扫一扫，把题目装进口袋