python使用NumPy计算两个向量之间的余弦相似度

2023-03-10 00:00:00 两个 向量 余弦

在 NumPy 中,可以使用 numpy.dot 和 numpy.linalg.norm 函数来计算两个向量的余弦相似度。余弦相似度是计算两个向量之间的相似度的一种方法。它测量两个非零向量之间夹角的余弦值,值在 -1 到 1 之间,其中 1 表示两个向量方向完全相同,-1 表示它们的方向完全相反。以下是一个简单的示例,演示如何使用 NumPy 计算两个向量之间的余弦相似度:

import numpy as np

# 定义两个向量
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# 计算余弦相似度
cos_sim = np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b))

print("向量 a 为:", a)
print("向量 b 为:", b)
print("余弦相似度为:", cos_sim)

在这个示例中,我们首先导入 NumPy 库,然后定义了两个向量 a 和 b。然后,我们使用 numpy.dot 函数和 numpy.linalg.norm 函数来计算这两个向量之间的余弦相似度。最后,我们将结果打印出来。

输出结果应该是:

向量 a 为: [1 2 3]
向量 b 为: [4 5 6]
余弦相似度为: 0.9746318461970762

因此,可以看到,使用 numpy.dot 和 numpy.linalg.norm 函数可以方便地在 NumPy 中计算两个向量之间的余弦相似度。

相关文章