python编程实现 Apriori算法的代码

很高兴为您解答问题！以下是 Python 实现的 Apriori 算法示例代码，希望您能够找到它对您的帮助和支持：

def loadDataSet():
    """简单的数据集"""
    return [[1, 3, 4], [2, 3, 5], [1, 2, 3, 5], [2, 5]]

def createC1(dataSet):
    """创建集合 C1，即对 dataSet 进行去重，排序，放入 frozenset 中，作为键值。不可改变"""
    C1 = []
    for transaction in dataSet:
        for item in transaction:
            if not [item] in C1:
                C1.append([item])

    C1.sort()
    return list(map(frozenset, C1))   # 使用 frozenset，将其作为字典的 key

def scanD(D, Ck, minSupport):
    """
    D：数据集
    Ck: ck项集
    minSupport: 最小支持率
    返回：
    retList：满足支持度的列表
    supportData：支持度字典
    """
    ssCnt = {}
    for tid in D:
        for can in Ck:
            if can.issubset(tid):
                if not can 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.insert(0,key)
        supportData[key] = support
    return retList, supportData

def aprioriGen(Lk, k): #Lk为频繁项集列表，k为生成的项集中所含有的元素个数
    """
    频繁项集列表 Lk，k —> Ck。
    由 L(k-1) × L(1) 生成 Ck。
    """
    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()
            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)) # 转换成 set 类型
    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

# 示例
dataSet = loadDataSet()
L, suppData = apriori(dataSet)
print("L:",L)
print("suppData:",suppData)

希望这可以回答您的第一个问题，如果您还有其他问题，请随时问我！