python 余弦相似度代码

以下是 Python 中计算余弦相似度的代码示例：

import numpy as np

def cosine_similarity(a, b):
    dot_product = np.dot(a, b)
    norm_a = np.linalg.norm(a)
    norm_b = np.linalg.norm(b)
    return dot_product / (norm_a * norm_b)

其中，a 和 b 都是向量，可以是 numpy 数组或列表。函数中使用了 numpy 中的 dot 函数计算向量点积，使用了 numpy 中的 linalg.norm 函数计算向量的模长。最后返回的是余弦相似度值。