讲师中心 微信公众号
AI工具推荐 视频效率加速

如何使用Python计算文本相似度—TF-IDF与余弦相似度

轻杰酱_6697

轻杰酱_6697

发布时间:2025-07-05 11:58:01

|

677人浏览过

|

来源于php中文网

原创

要比较两段文本的相似程度,使用tf-idf结合余弦相似度是一种常见且有效的方法。tf-idf用于评估词语在文档中的重要性,由词频(tf)和逆文档频率(idf)组成;余弦相似度通过计算向量夹角的余弦值衡量相似性,值越接近1表示越相似。实现流程为:1. 使用jieba进行中文分词;2. 利用tfidfvectorizer将文本转为tf-idf向量;3. 通过cosine_similarity函数计算相似度。注意事项包括:分词工具影响结果准确性、需处理停用词、文本长度差异可通过预处理解决。此外,批量比较多个文本时可一次性计算所有组合的相似度。该方法虽非最先进,但实现简单且效果良好。

如何使用Python计算文本相似度—TF-IDF与余弦相似度

要比较两段文本的相似程度,TF-IDF结合余弦相似度是一个常见且有效的方法。它能将文本转化为向量,并通过数学方式衡量它们之间的“距离”或“相似性”。

如何使用Python计算文本相似度—TF-IDF与余弦相似度

什么是TF-IDF和余弦相似度?

TF-IDF(Term Frequency-Inverse Document Frequency) 是一种统计方法,用于评估一个词在文档中的重要程度。简单来说:

如何使用Python计算文本相似度—TF-IDF与余弦相似度
  • TF(词频):某个词在文档中出现的次数。
  • IDF(逆文档频率):衡量这个词在整个语料库中的普遍重要性,越少见的词IDF越高。

余弦相似度(Cosine Similarity) 则是通过计算两个向量夹角的余弦值来判断它们的相似程度。值越接近1,表示越相似。

立即学习“Python免费学习笔记(深入)”;

所以整个流程大致是:文本 → TF-IDF向量化 → 余弦相似度计算。

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
如何使用Python计算文本相似度—TF-IDF与余弦相似度

怎么用Python实现?

Python有很多现成的库可以完成这项任务,比如 scikit-learn 和 jieba(中文分词)。下面是一个基本流程:

1. 安装必要库

pip install scikit-learn jieba

2. 分词处理(中文)

对于英文可以直接按空格分割,中文需要使用像 jieba 这样的分词工具:

import jieba

text1 = "我喜欢机器学习"
text2 = "我热爱人工智能"

words1 = " ".join(jieba.cut(text1))
words2 = " ".join(jieba.cut(text2))

3. 使用 TfidfVectorizer 转换为向量

from sklearn.feature_extraction.text import TfidfVectorizer

corpus = [words1, words2]

vectorizer = TfidfVectorizer()
tfidf_matrix = vectorizer.fit_transform(corpus)

4. 计算余弦相似度

from sklearn.metrics.pairwise import cosine_similarity

similarity = cosine_similarity(tfidf_matrix[0], tfidf_matrix[1])
print(f"相似度:{similarity[0][0]:.4f}")

常见问题与注意事项

分词对结果影响大

  • 中文必须先分词,否则直接当作一个个字处理会严重影响效果。
  • 不同的分词工具(如jieba、HanLP)可能会有不同的切分结果,影响最终相似度。

停用词处理

一些无意义的词(如“的”、“了”、“是”)应该被过滤掉。可以用停用词表提升准确性:

vectorizer = TfidfVectorizer(stop_words=your_stopword_list)

文本长度差异太大怎么办?

如果两段文本长度相差悬殊,可能会影响TF-IDF的分布。可以考虑先做预处理,比如截断或分块比较。


小技巧:批量比较多个文本

如果你有多个文本需要比较,可以把所有文本都转换成TF-IDF向量,然后一次性计算所有组合的余弦相似度:

from sklearn.metrics.pairwise import cosine_similarity

all_texts = [text1, text2, text3]  # 更多文本
corpus = [" ".join(jieba.cut(t)) for t in all_texts]

tfidf_matrix = vectorizer.fit_transform(corpus)
similarities = cosine_similarity(tfidf_matrix)

# similarities[i][j] 就是第i篇和第j篇的相似度

基本上就这些。TF-IDF + 余弦相似度虽然不是最先进的方法,但在很多实际场景下已经够用了,而且实现起来不难。只要注意分词和停用词这些细节,就能得到不错的匹配结果。

热门AI工具

更多
AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

相关专题

更多
Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

80

2026.10.22

Gradle Wrapper配置教程
Gradle Wrapper配置教程

本专题介绍Gradle Wrapper的基本作用,涵盖文件组成、生成方法、版本配置和任务调用,帮助你理解项目如何使用指定的Gradle版本,减少不同开发环境带来的构建差异。

40

2026.10.22

Golangci-lint检查器使用教程
Golangci-lint检查器使用教程

本专题整理Golangci-lint常用Linters的用途与配置方法,覆盖错误处理、无效代码、安全、复杂度、命名和依赖检查,帮助项目合理选规则。

40

2026.10.22

Gradle怎么构建Java项目
Gradle怎么构建Java项目

本专题介绍Gradle Java项目构建方法,涵盖Java插件配置、依赖声明、源码编译、测试执行和打包检查,帮助你理解各阶段的任务关系,逐步完成项目构建。

20

2026.10.22

PHP时间戳漏洞修复教程
PHP时间戳漏洞修复教程

本专题汇总PHP常见时间戳漏洞,包括2038年溢出、时区漂移、时间戳伪造、重放攻击、时间盲注与日志审计失效等,并给出升级64位环境、使用 DateTimeImmutable、统一UTC、严格超时校验、预处理绑定、hash_equals 比对等修复方案,帮助开发者快速排查和加固时间相关代码。

40

2026.10.22

Golangci-lint配置文件与规则设置
Golangci-lint配置文件与规则设置

本专题聚焦Golangci-lint YAML配置文件,讲清运行参数、规则启停、目录排除、问题过滤及严重级别设置,便于团队统一检查标准。

20

2026.10.22

Golangci-lint安装与入门教程
Golangci-lint安装与入门教程

本专题讲解Golangci-lint的安装、初始化、基础命令和首次代码检查,覆盖本地环境、Docker及多平台用法,帮助Go开发者快速上手。

40

2026.10.22

PHP2Shell高危漏洞修复教程
PHP2Shell高危漏洞修复教程

PHP2Shell漏洞是什么,哪些PHP版本和运行环境会受到影响,服务器应该如何检测与修复?本专题提供版本自查脚本、Linux与Docker升级命令、PHP-FPM和Apache排查步骤,以及修复验证、临时缓解和应急响应方案。

240

2026.10.22

剧云人物设定与关系管理教程
剧云人物设定与关系管理教程

本专题介绍剧云人物设定与关系管理,讲解角色小传、性格职业、人物目标、背景资料和关系网络的创建及维护方法。

20

2026.10.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn