【数据挖掘实战】从购物篮到推荐系统:Apriori与FP-Growth算法Python全解析

📅 2026/7/15 21:55:21
【数据挖掘实战】从购物篮到推荐系统:Apriori与FP-Growth算法Python全解析
1. 从啤酒与尿布说起频繁模式挖掘的商业价值1990年代沃尔玛的工程师在分析销售数据时发现一个有趣现象啤酒和尿布经常出现在同一张购物小票上。经过调查发现年轻父亲们常在购买婴儿用品时顺手带走几罐啤酒。这个发现让沃尔玛调整货架布局将啤酒和尿布陈列在相邻区域销售额因此显著提升。这个经典案例揭示了购物篮分析Market Basket Analysis的魔力——通过挖掘商品之间的关联关系可以优化销售策略。现代推荐系统、广告投放、库存管理等场景都依赖这项技术。比如电商平台猜你喜欢功能外卖App的套餐搭配建议超市的促销商品组合设计频繁模式指在数据集中反复出现的组合可以是项集{牛奶,面包}同时出现子序列用户先浏览手机再查看保护壳子结构社交网络中的常见互动模式2. 核心概念与评估指标2.1 关键术语解析事务(Transaction)一组物品的集合如一次购物记录项集(Itemset)若干物品的集合如{牛奶,面包}支持度(Support)项集出现的频率计算式为support(X) count(X) / total_transactions置信度(Confidence)规则X→Y的可信程度confidence(X→Y) support(X∪Y) / support(X)2.2 算法目标找出所有满足最小支持度的频繁项集从频繁项集中提取置信度高于阈值的强关联规则2.3 示例计算假设有以下5条交易记录1: 牛奶,面包 2: 牛奶,尿布,啤酒 3: 牛奶,尿布,面包 4: 尿布,啤酒 5: 面包,鸡蛋计算{牛奶,尿布}的支持度出现次数2次(交易2和3)总交易数5支持度 2/5 40%计算规则牛奶→尿布的置信度support(牛奶,尿布) 2/5support(牛奶) 3/5置信度 (2/5)/(3/5) ≈ 66.7%3. Apriori算法详解3.1 算法原理Apriori基于一个关键性质如果一个项集是频繁的那么它的所有子集也是频繁的。反之若某子集不频繁则包含它的超集也不频繁。算法采用逐层搜索的迭代方法扫描数据库找出频繁1项集L₁用L₁生成候选2项集C₂筛选得到L₂用L₂生成C₃依此类推直到无法生成更大项集3.2 Python实现3.2.1 使用现成库from efficient_apriori import apriori transactions [(牛奶,面包), (牛奶,尿布,啤酒), (牛奶,尿布,面包), (尿布,啤酒), (面包,鸡蛋)] itemsets, rules apriori(transactions, min_support0.4, min_confidence0.6) print(rules) # 输出关联规则3.2.2 从零实现def create_C1(data): 生成候选1项集 C1 [] for transaction in data: for item in transaction: if [item] not in C1: C1.append([item]) return list(map(frozenset, C1)) def scan_D(D, Ck, min_support): 筛选满足支持度的项集 ss_cnt {} for tid in D: for can in Ck: if can.issubset(tid): ss_cnt[can] ss_cnt.get(can,0) 1 num_items float(len(D)) ret_list [] support_data {} for key in ss_cnt: support ss_cnt[key]/num_items if support min_support: ret_list.insert(0,key) support_data[key] support return ret_list, support_data3.3 优缺点分析优势原理简单直观适合小规模数据集缺陷多次扫描数据库I/O开销大候选项集数量可能爆炸式增长对长模式效率低下4. FP-Growth算法进阶4.1 算法思想FP-Growth通过构建FP树压缩数据只需扫描数据库两次第一次扫描统计项频次过滤非频繁项第二次扫描构建FP树从FP树挖掘频繁模式4.2 FP树结构示例对于之前的交易数据按频次降序排列后尿布:4, 牛奶:3, 面包:3, 啤酒:2, 鸡蛋:1构建的FP树如下数字表示出现次数Null ├─ 尿布:4 │ ├─ 牛奶:2 │ │ ├─ 面包:1 │ │ └─ 啤酒:1 │ └─ 啤酒:1 └─ 牛奶:1 └─ 面包:14.3 Python实现class TreeNode: FP树节点类 def __init__(self, name, count, parent): self.name name self.count count self.parent parent self.children {} self.link None # 相似项链接 def create_tree(dataset, min_sup1): 构建FP树 header_table {} # 第一次扫描统计频次 for trans in dataset: for item in trans: header_table[item] header_table.get(item,0) dataset[trans] # 移除不满足最小支持度的项 for k in list(header_table.keys()): if header_table[k] min_sup: del header_table[k] freq_items set(header_table.keys()) if not freq_items: return None, None # 初始化头指针表 for k in header_table: header_table[k] [header_table[k], None] # 构建树 root TreeNode(Null, 1, None) for items, count in dataset.items(): localD {} for item in items: if item in freq_items: localD[item] header_table[item][0] if localD: ordered_items [v[0] for v in sorted(localD.items(), keylambda x: x[1], reverseTrue)] update_tree(ordered_items, root, header_table, count) return root, header_table4.4 性能对比指标AprioriFP-Growth扫描次数多次2次速度慢快5-10倍内存占用较低较高适合场景小数据集大数据集5. 实战构建推荐系统5.1 数据准备使用MovieLens电影评分数据集import pandas as pd ratings pd.read_csv(ratings.csv) movies pd.read_csv(movies.csv) # 将评分转换为二元数据评分3视为喜欢 ratings[liked] ratings[rating] 3 user_movies ratings[ratings[liked]].groupby(userId)[movieId].apply(list)5.2 关联规则挖掘from mlxtend.preprocessing import TransactionEncoder from mlxtend.frequent_patterns import apriori, association_rules # 数据编码 te TransactionEncoder() te_ary te.fit(user_movies).transform(user_movies) df pd.DataFrame(te_ary, columnste.columns_) # 挖掘频繁项集 frequent_itemsets apriori(df, min_support0.1, use_colnamesTrue) # 提取关联规则 rules association_rules(frequent_itemsets, metriclift, min_threshold1.2) rules rules.sort_values([lift, confidence], ascending[False, False])5.3 推荐实现def recommend(movie_id, rules, top_n5): 根据关联规则推荐电影 related rules[rules[antecedents].apply(lambda x: movie_id in x)] recommendations [] for _, row in related.iterrows(): for movie in row[consequents]: if movie not in recommendations: recommendations.append(movie) if len(recommendations) top_n: return recommendations return recommendations6. 算法选择与优化技巧6.1 选择依据数据规模小数据用Apriori大数据用FP-Growth模式长度长模式优先FP-Growth实时性要求FP-Growth更适合实时系统6.2 性能优化数据预处理离散化连续变量合并稀疏类别移除高频常见项如购物袋参数调优# 动态调整支持度阈值 def adaptive_support(current_iter): return max(0.1, 0.5 - 0.05*current_iter)并行计算from pyspark.mllib.fpm import FPGrowth data sc.textFile(data.txt).map(lambda line: line.strip().split(,)) model FPGrowth.train(data, minSupport0.2, numPartitions10)7. 扩展应用场景7.1 异常检测频繁模式可识别正常行为模式偏离这些模式的即为异常信用卡欺诈检测网络入侵识别工业设备故障预警7.2 序列模式挖掘分析时间序列中的频繁模式from prefixspan import PrefixSpan ps PrefixSpan([ [C,A,B], [A,B,D], [B,A,C] ]) ps.minlen 2 print(ps.frequent(2)) # 最小支持度为27.3 图模式挖掘在复杂网络中寻找频繁子图社交网络中的常见互动模式分子结构中的功能基团识别交通网络中的热点路径实际项目中我曾在用户行为分析中使用FP-Growth算法从千万级事件日志中挖掘出高频操作序列帮助产品团队优化了工作流设计使关键路径的完成率提升了18%。需要注意的是算法结果需要结合业务知识解读——有时高置信度规则可能是数据偏差所致而非真实关联。