详解Python 关联规则分析

时间:2022-11-09 20:36:18

1. 关联规则

大家可能听说过用于宣传数据挖掘的一个案例:啤酒和尿布;据说是沃尔玛超市在分析顾客的购买记录时,发现许多客户购买啤酒的同时也会购买婴儿尿布,于是超市调整了啤酒和尿布的货架摆放,让这两个品类摆放在一起;结果这两个品类的销量都有明显的增长;分析原因是很多刚生小孩的男士在购买的啤酒时,会顺手带一些婴幼儿用品。

不论这个案例是否是真实的,案例中分析顾客购买记录的方式就是关联规则分析法association rules。

关联规则分析也被称为购物篮分析,用于分析数据集各项之间的关联关系。

1.1 基本概念

  • 项集:item的集合,如集合{牛奶、麦片、糖}是一个3项集,可以认为是购买记录里物品的集合。
  • 频繁项集:顾名思义就是频繁出现的item项的集合。如何定义频繁呢?用比例来判定,关联规则中采用支持度和置信度两个概念来计算比例值
  • 支持度:共同出现的项在整体项中的比例。以购买记录为例子,购买记录100条,如果商品a和b同时出现50条购买记录(即同时购买a和b的记录有50),那边a和b这个2项集的支持度为50%

详解Python 关联规则分析

  • 置信度:购买a后再购买b的条件概率,根据贝叶斯公式,可如下表示:

详解Python 关联规则分析

提升度:为了判断产生规则的实际价值,即使用规则后商品出现的次数是否高于商品单独出现的评率,提升度和衡量购买x对购买y的概率的提升作用。如下公式可见,如果x和y相互独立那么提升度为1,提升度越大,说明x->y的关联性越强

详解Python 关联规则分析

1.2 关联规则apriori算法

关联规则方法的步骤如下:

  • 发现频繁项集
  • 找出关联规则

apriori算法是经典的关联规则算法。apriori算法的目标是找到最大的k项频繁集。apriori算法从寻找1项集开始,通过最小支持度阈值进行剪枝,依次寻找2项集,3项集直到没有更过项集为止。

下面是一个案例图解:

详解Python 关联规则分析

  • 图中有4个记录,记录项有1,2,3,4,5若干
  • 首先先找出1项集对应的支持度(c1),可以看出4的支持度低于最小支持阈值,先剪掉(l1)。
  • 从1项集生成2项集,并计算支持度(c2),可以看出(1,5)(1,2)支持度低于最小支持阈值,先剪掉(l2)
  • 从2项集生成3项集,(1,2,3)(1,2,5)(2,3,5)只有(2,3,5)满足要求
  • 没有更多的项集了,就定制迭代

2. mlxtend实战关联规则

关联规则目前在scikit-learn中并没有实现。这里介绍另一个python库mlxtend。

2.1 安装

?
1
pip install mlxtend

2.2 简单的例子

来看下数据集:

?
1
2
3
4
5
6
7
8
9
import pandas as pd
 
item_list = [['牛奶','面包'],
    ['面包','尿布','啤酒','土豆'],
    ['牛奶','尿布','啤酒','可乐'],
    ['面包','牛奶','尿布','啤酒'],
    ['面包','牛奶','尿布','可乐']]
 
item_df = pd.dataframe(item_list)

数据格式处理,传入模型的数据需要满足bool值的格式

?
1
2
3
4
5
from mlxtend.preprocessing import transactionencode
 
te = transactionencoder()
df_tf = te.fit_transform(item_list)
df = pd.dataframe(df_tf,columns=te.columns_)

详解Python 关联规则分析

  • 计算频繁项集
?
1
2
3
4
5
6
7
8
9
from mlxtend.frequent_patterns import apriori
 
# use_colnames=true表示使用元素名字,默认的false使用列名代表元素, 设置最小支持度min_support
frequent_itemsets = apriori(df, min_support=0.05, use_colnames=true)
 
frequent_itemsets.sort_values(by='support', ascending=false, inplace=true)
 
# 选择2频繁项集
print(frequent_itemsets[frequent_itemsets.itemsets.apply(lambda x: len(x)) == 2])  

详解Python 关联规则分析

  • 计算关联规则
?
1
2
3
4
5
6
7
8
9
from mlxtend.frequent_patterns import association_rules
 
# metric可以有很多的度量选项,返回的表列名都可以作为参数
association_rule = association_rules(frequent_itemsets,metric='confidence',min_threshold=0.9)
 
#关联规则可以提升度排序
association_rule.sort_values(by='lift',ascending=false,inplace=true)    
association_rule
# 规则是:antecedents->consequents

详解Python 关联规则分析

选择出来关联规则之后,根据提升度排序后,可能最高提升度的规则是在我们常识范围内,那这个规则的价值就不高。所以我们要在产生的规则中根据业务特点进行筛选,像开篇提到(啤酒->尿布)完全不同的品类之间的关联。

笔者最近用关联规则分析用户的体检报告记录,也得出了关于各个病症的有意义的关联,如并发症,不同病症相互影响等。

3. 总结

本分介绍关联规则的基本概念和经典算法apriori,以及python的实现库mlxtend使用。

总结如下:

  • 关联规则用于分析数据集各项之间的关联关系,想一想啤酒和尿布的故事
  • 三个重要概念:支持度,置信度和提升度
  • apriori通过迭代先找1项集,用支持度过滤项集,逐步找出所有k项集
  • 用置信度或提升度来选择满足的要求的规则
  • mlxtend对数据要求转换成bool值才可用

以上就是详解python 关联规则分析的详细内容,更多关于python 关联规则分析的资料请关注服务器之家其它相关文章!

原文链接:https://juejin.cn/post/6940506066513821703