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

Python推荐系统 Python协同过滤算法实现步骤

雨明酱_9611

雨明酱_9611

发布时间:2025-06-08 21:18:02

|

721人浏览过

|

来源于php中文网

原创

协同过滤推荐系统可通过以下步骤用python实现:1. 数据准备:获取用户-物品评分数据,如movielens等;2. 构建用户-物品矩阵,使用pandas的pivot方法转换数据结构;3. 计算相似度,基于用户或物品,常用余弦相似度或皮尔逊相关系数;4. 预测评分并生成推荐,通过加权平均相似用户评分得出候选列表;5. 注意事项包括冷启动问题、评分标准化、稀疏性优化及性能优化。每一步均需结合代码实现并根据实际场景调整策略以提升效果。

Python推荐系统 Python协同过滤算法实现步骤

推荐系统是现在很多平台的核心功能之一,比如电商、视频网站、音乐平台等。Python 作为数据处理和算法开发的主力语言,在构建推荐系统方面非常合适。协同过滤(Collaborative Filtering)是最经典的推荐算法之一,实现起来也不算太难。

下面我来一步步说说怎么用 Python 实现一个简单的协同过滤推荐系统。


1. 数据准备:用户-物品评分矩阵

协同过滤的基础是用户对物品的评分数据。最常见的是“用户-物品-评分”的三列结构,比如:

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

user_id | item_id | rating
--------|---------|-------
1       | A       | 5
1       | B       | 3
2       | A       | 4
...

你可以使用公开的数据集,例如 MovieLens,也可以自己整理类似格式的数据。

常见问题:评分缺失太多怎么办? 答案很简单:没关系,协同过滤本来就是用来处理稀疏矩阵的。

2. 构建用户-物品评分矩阵

有了原始数据之后,第一步是把它转成一个二维矩阵,行表示用户,列表示物品,值是评分。

可以用 Pandas 轻松完成这一步:

import pandas as pd

# 假设你已经加载了数据到 df,包含 user_id, item_id, rating
ratings_matrix = df.pivot(index='user_id', columns='item_id', values='rating')

这时候你会得到一个看起来像这样的矩阵:

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

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

下载
item_id     A    B    C
user_id                 
1         5.0  3.0  NaN
2         4.0  NaN  2.0
3         NaN  4.0  5.0

3. 计算相似度:用户之间 or 物品之间?

协同过滤分为两种:

  • 基于用户的协同过滤(User-CF):找和当前用户兴趣相近的其他用户,然后推荐他们喜欢但当前用户没看过的东西。
  • 基于物品的协同过滤(Item-CF):找和目标物品相似的其他物品,推荐给喜欢过这些相似物品的用户。

这里以 User-CF 为例,计算用户之间的相似度常用方法是皮尔逊相关系数或余弦相似度。

from sklearn.metrics.pairwise import cosine_similarity

# 填充 NaN 为 0,方便计算相似度
similarity_matrix = cosine_similarity(ratings_matrix.fillna(0))

这样你就得到了一个用户之间的相似度矩阵。


4. 预测评分并生成推荐

接下来就是预测某个用户对未评分物品的兴趣程度。公式大致如下:

$$ \hat{r}_{ui} = \bar{r}u + \frac{\sum{v \in N(u)} \text{sim}(u,v) \cdot (r_{vi} - \bar{r}v)}{\sum{v \in N(u)} |\text{sim}(u,v)|} $$

不过在实际代码中可以简化处理,比如先找出最相似的几个用户,再根据他们的评分加权平均。

举个例子:

# 找出与用户1最相似的前3个用户
similar_users = similarity_df[1].sort_values(ascending=False)[1:4].index

# 收集这些用户评价过的电影,但用户1没有看过的
candidate_items = ratings_matrix.loc[similar_users].mean(axis=0).sort_values(ascending=False)

# 排除用户1已经评过分的项目
user_rated = ratings_matrix.loc[1].dropna().index
recommendations = candidate_items.drop(user_rated, errors='ignore')

最终 recommendations 就是你想推荐的内容。


5. 注意事项 & 小技巧

  • 冷启动问题:新用户或新物品没有评分,无法推荐。这是协同过滤的硬伤,只能配合其他策略缓解。
  • 评分标准化:有些用户打分偏高,有些偏低,建议做去中心化处理(比如减去用户的平均评分)。
  • 稀疏性优化:矩阵太大太稀疏时,可以用稀疏矩阵(scipy.sparse)节省内存。
  • 性能优化:如果用户/物品太多,直接两两比较会很慢,可以考虑使用近似最近邻(ANN)加速。

基本上就这些步骤了。协同过滤虽然简单,但效果不错,尤其适合入门推荐系统。如果你能结合一些内容特征或者上下文信息,效果还能更上一层楼。

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

DeepSeek

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

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

UP简历
UP简历 Hot

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

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

WorkBuddy

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

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

豆包大模型

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

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

340

2025.12.04

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

252

2026.01.31

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

691

2026.04.13

treenode的用法
treenode的用法

​在计算机编程领域,TreeNode是一种常见的数据结构,通常用于构建树形结构。在不同的编程语言中,TreeNode可能有不同的实现方式和用法,通常用于表示树的节点信息。更多关于treenode相关问题详情请看本专题下面的文章。php中文网欢迎大家前来学习。

2681

2023.12.01

C++ 高效算法与数据结构
C++ 高效算法与数据结构

本专题讲解 C++ 中常用算法与数据结构的实现与优化,涵盖排序算法(快速排序、归并排序)、查找算法、图算法、动态规划、贪心算法等,并结合实际案例分析如何选择最优算法来提高程序效率。通过深入理解数据结构(链表、树、堆、哈希表等),帮助开发者提升 在复杂应用中的算法设计与性能优化能力。

376

2025.12.22

深入理解算法:高效算法与数据结构专题
深入理解算法:高效算法与数据结构专题

本专题专注于算法与数据结构的核心概念,适合想深入理解并提升编程能力的开发者。专题内容包括常见数据结构的实现与应用,如数组、链表、栈、队列、哈希表、树、图等;以及高效的排序算法、搜索算法、动态规划等经典算法。通过详细的讲解与复杂度分析,帮助开发者不仅能熟练运用这些基础知识,还能在实际编程中优化性能,提高代码的执行效率。本专题适合准备面试的开发者,也适合希望提高算法思维的编程爱好者。

417

2026.01.06

C++ 数据结构与算法实现教程合集
C++ 数据结构与算法实现教程合集

以 C++ 为实现语言,系统讲解核心数据结构与算法,涵盖链表(单链表/双链表/环检测)、栈与队列(单调栈/优先队列)、二叉树(遍历/BST/AVL/红黑树)、哈希表(开地址法/链地址法)、图(邻接表/BFS/DFS/Dijkstra/拓扑排序)、常见排序算法(快排/归并/堆排/计数排序)的实现与复杂度分析,同时分享 LeetCode 刷题技巧、竞赛编程常用模板(二分/前缀和/滑动窗口/动态规划),帮助开发者夯实算法基础。

472

2026.05.09

页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

6376

2023.08.14

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

40

2026.10.28

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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