使用Apriori算法进行关联分析(一)

概述

  • 优点:易编码实现
  • 缺点:在大数据集上可能较慢
  • 适用数据类型:数值型或标称型数据

关联分析

从大规模数据集中寻找物品间的隐含关系被称作关联分析(association analysis)或者关联规则学习(association rule learning)

这些隐含关系可以有两种形式:频繁项集和关联规则。

  • 频繁项集(frequent item sets):经常出现在一起的物品的集合。
  • 关联规则(association rules):暗示两种物品之间可能存在很强的关系。

可以用支持度和可信度来衡量定义这种关系。
一个项集的支持度(support)被定义为数据集中包含该项集的记录所占的比例。
可信度或置信度(confidence)是针对一条诸如{尿布} -> {葡萄酒}的关联规则来定义的。这条规则的可信度被定义为支持度({尿布,葡萄酒})/支持度({尿布})

Apriori原理

假设我们只有4种商品在卖,商品0,商品1,商品2,商品3。顾客可能购买的商品组合如下图所示。


可能的商品组合

我们的目标是找到经常在一起购买的物品集合,使用支持度来度量其出现的频率。例如计算{0,3}的支持度,我们需要遍历所有记录,找到同时包含0和3的记录,就将计数值加1,最后用计数值除以总记录数。

对于我们只有4种商品的集合,需要遍历数据集15次(观察上图一共有15种可能组合),对于包含N种物品的数据集一共有2^N-1种组合。计算量相当庞大。

我们可以用Apriori原理来减少项集,降低所需计算时间。

Apriori原理是说如果某个项集是频繁的,那么它的所有子集也是频繁的。反过来说就是,如果一个项集是非频繁的,那么它的所有超集也是非频繁的。


例如我们计算{2,3}的支持度,知道它是非频繁的,那么就不再需要计算{0,2,3}、{1,2,3}和{0,1,2,3}了。

使用Apriori原理可以避免项集数目的指数增长,从而在合理时间内计算出频繁项集。

使用Apriori算法发现频繁集

生成候选项集

# 构建数据集
def createData():
    return [[1, 3, 4], [2, 3, 5], [1, 2, 3, 5], [2, 5]]

# 构建C1候选项集
def createC1(dataSet):
    C1 = []
    for transaction in dataSet:
        for item in transaction:
            if [item] not in C1:
                C1.append([item])
    C1.sort()
    return list(map(frozenset, C1)) 

def scanD(D, Ck, minSupport):
    ssCnt = {}
    # 遍历所有记录,计数各项集出现次数
    for tid in D:
        for can in Ck:
            if can.issubset(tid):
                if can not in ssCnt: 
                    ssCnt[can] = 1
                else: 
                    ssCnt[can] += 1
    numItems = float(len(D))
    retList = []
    supportData = {}
    for key in ssCnt:
        support = ssCnt[key]/numItems   # 计算支持度
        if support >= minSupport:
            retList.append(key)
        supportData[key] = support
    return retList, supportData

函数createC1()构建大小为1的所有候选项集的集合C1。Apriori算法首先构建集合C1,然后扫描数据集来找到满足最小支持度的频繁项集构成集合L1,而L1中的元素相互组合构成C2,C2再进一步过滤得到L2,以此类推。

所以Apriori算法需要函数createC1()来构建第一个候选项集的列表C1。由于列表中的项集后面需要被当作字典的key,所以返回时用map()函数将各项集转化为frozenset类型。

scanD()函数用于从Ck生成Lk,另外还会返回一个包含支持度值的字典以备后用。

下面看下实际运行效果。

data = createData()
C1 = createC1(data)
D = list(map(set, data))
L1, supportData = scanD(D, C1, 0.5)

第一个候选项集C1,数据集D和L1的值如下。


完整的Apriori算法

整个Apriori算法的伪代码如下:

当集合中项的个数大于0时:
    构建一个k个项组成的候选项集的列表
    检查数据以确认每个项集都是频繁的
    保留频繁项集并构建k+1项组成的候选项集的列表

代码如下:

def aprioriGen(Lk, k):
    retList = []
    lenLk = len(Lk)
    for i in range(lenLk):
        for j in range(i+1, lenLk):
            L1 = list(Lk[i])[:k-2]
            L2 = list(Lk[j])[:k-2]
            L1.sort()
            L2.sort()
            # 前k-2项相同时,将两个项集合并
            if L1 == L2:
                retList.append(Lk[i] | Lk[j])
    return retList

def apriori(dataSet, minSupport = 0.5):
    C1 = createC1(dataSet)
    D = list(map(set, dataSet))
    L1, supportData = scanD(D, C1, minSupport)
    L = [L1]
    k = 2
    while (len(L[k-2]) > 0):
        Ck = aprioriGen(L[k-2], k)
        Lk, supK = scanD(D, Ck, minSupport)
        supportData.update(supK)
        L.append(Lk)
        k += 1
    return L, supportData

函数aprioriGen()的输入参数为频繁项集列表Lk与项集元素个数k,输出为Ck。

apriori()为主函数。首先调用createC1()创建第一个候选项集,然后scanD()生成L1,接着循环调用aprioriGen()scanD()生成Ck和Lk,最后当Lk为空时退出。

下面是执行效果。


这是最小支持度为50%的情况,下面试一下75%的支持度。


下面画出的简陋流程图用于加深对Apriori算法找出频繁集的工作流程的理解。


Apriori算法流程简易版

到此,我们就用Apriori算法找出频繁集了。后面就是要从频繁项集中挖掘关联规则


都看到最后了,要不~点个赞?加波关注?

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,496评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,407评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,632评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,180评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,198评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,165评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,052评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,910评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,324评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,542评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,711评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,424评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,017评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,668评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,823评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,722评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,611评论 2 353

推荐阅读更多精彩内容