0

0

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

星夢妙者

星夢妙者

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

|

577人浏览过

|

来源于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向量化 → 余弦相似度计算。

Kacha
Kacha

KaCha是一款革命性的AI写真工具,用AI技术将照片变成杰作!

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

怎么用Python实现?

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

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工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

16

2026.03.11

Go高并发任务调度与Goroutine池化实践
Go高并发任务调度与Goroutine池化实践

本专题围绕 Go 语言在高并发任务处理场景中的实践展开,系统讲解 Goroutine 调度模型、Channel 通信机制以及并发控制策略。内容包括任务队列设计、Goroutine 池化管理、资源限制控制以及并发任务的性能优化方法。通过实际案例演示,帮助开发者构建稳定高效的 Go 并发任务处理系统,提高系统在高负载环境下的处理能力与稳定性。

23

2026.03.10

Kotlin Android模块化架构与组件化开发实践
Kotlin Android模块化架构与组件化开发实践

本专题围绕 Kotlin 在 Android 应用开发中的架构实践展开,重点讲解模块化设计与组件化开发的实现思路。内容包括项目模块拆分策略、公共组件封装、依赖管理优化、路由通信机制以及大型项目的工程化管理方法。通过真实项目案例分析,帮助开发者构建结构清晰、易扩展且维护成本低的 Android 应用架构体系,提升团队协作效率与项目迭代速度。

75

2026.03.09

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

95

2026.03.06

Rust内存安全机制与所有权模型深度实践
Rust内存安全机制与所有权模型深度实践

本专题围绕 Rust 语言核心特性展开,深入讲解所有权机制、借用规则、生命周期管理以及智能指针等关键概念。通过系统级开发案例,分析内存安全保障原理与零成本抽象优势,并结合并发场景讲解 Send 与 Sync 特性实现机制。帮助开发者真正理解 Rust 的设计哲学,掌握在高性能与安全性并重场景中的工程实践能力。

218

2026.03.05

PHP高性能API设计与Laravel服务架构实践
PHP高性能API设计与Laravel服务架构实践

本专题围绕 PHP 在现代 Web 后端开发中的高性能实践展开,重点讲解基于 Laravel 框架构建可扩展 API 服务的核心方法。内容涵盖路由与中间件机制、服务容器与依赖注入、接口版本管理、缓存策略设计以及队列异步处理方案。同时结合高并发场景,深入分析性能瓶颈定位与优化思路,帮助开发者构建稳定、高效、易维护的 PHP 后端服务体系。

420

2026.03.04

AI安装教程大全
AI安装教程大全

2026最全AI工具安装教程专题:包含各版本AI绘图、AI视频、智能办公软件的本地化部署手册。全篇零基础友好,附带最新模型下载地址、一键安装脚本及常见报错修复方案。每日更新,收藏这一篇就够了,让AI安装不再报错!

168

2026.03.04

Swift iOS架构设计与MVVM模式实战
Swift iOS架构设计与MVVM模式实战

本专题聚焦 Swift 在 iOS 应用架构设计中的实践,系统讲解 MVVM 模式的核心思想、数据绑定机制、模块拆分策略以及组件化开发方法。内容涵盖网络层封装、状态管理、依赖注入与性能优化技巧。通过完整项目案例,帮助开发者构建结构清晰、可维护性强的 iOS 应用架构体系。

222

2026.03.03

C++高性能网络编程与Reactor模型实践
C++高性能网络编程与Reactor模型实践

本专题围绕 C++ 在高性能网络服务开发中的应用展开,深入讲解 Socket 编程、多路复用机制、Reactor 模型设计原理以及线程池协作策略。内容涵盖 epoll 实现机制、内存管理优化、连接管理策略与高并发场景下的性能调优方法。通过构建高并发网络服务器实战案例,帮助开发者掌握 C++ 在底层系统与网络通信领域的核心技术。

33

2026.03.03

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号