python 余弦相似度
时间: 2023-06-21 10:09:36 浏览: 106
python 余弦相似度算法
余弦相似度是一种衡量两个向量之间相似度的方法,可以用于文本相似度的计算。在 Python 中,可以使用 NumPy 来计算余弦相似度。具体实现步骤如下:
1. 导入 NumPy 库
```python
import numpy as np
```
2. 定义两个向量
```python
vector1 = np.array([1, 2, 3])
vector2 = np.array([2, 3, 4])
```
3. 计算两个向量的余弦相似度
```python
cos_sim = np.dot(vector1, vector2) / (np.linalg.norm(vector1) * np.linalg.norm(vector2))
```
其中,`np.dot(vector1, vector2)` 表示向量点积,`np.linalg.norm(vector1)` 表示向量的模长。
4. 输出结果
```python
print(cos_sim)
```
输出结果为:
```
0.9922778767136677
```
表示两个向量的余弦相似度为 0.9922778767136677。
阅读全文