python计算余弦相似度

计算余弦相似度可以使用Python的NumPy库中的cos函数。假设有两个向量a和b，可以使用以下代码计算它们的余弦相似度：

import numpy as np

# 定义两个向量a和b
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(cos_sim)

输出结果为：

0.9746318461970762

其中，`np.dot(a, b)`表示向量a和向量b的点积，`np.linalg.norm(a)`表示向量a的模长，`np.linalg.norm(b)`表示向量b的模长。